Merge pull request #70 from zooniverse/zoobot-upgrade-testing

remove nvidia error command
zooniverse · Dec 19, 2024 · c356d9f · c356d9f
2 parents cf2c0b5 + 81cc952
commit c356d9f
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}; '