remove gpu all nvidia error command

zooniverse · Dec 19, 2024 · 81cc952 · 81cc952
1 parent adec73a
commit 81cc952
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bajor/batch/train_finetuning.py b/bajor/batch/train_finetuning.py
@@ -234,7 +234,7 @@ def create_job_tasks(job_id, task_id=1, run_opts=''):
        '/bin/bash -c "'
        'set -ex; '
        f'{wait_for_preparation_task_completion}; '
-       'nvidia-smi; || echo \\"nvidia-smi command failed\\"; '
+       'nvidia-smi; '
        'python -c \\"import torch; print(torch.cuda.is_available()); print(torch.cuda.device_count())\\"; '
        f'{setup_pytorch_kernel_cache_env_var}; '
        f'{setup_hugging_face_cache_env_var}; '