Thank you for the reply. I tried setting CUDA_VISIBLE_DEVICES to both 0 and 0,1,3,4 and I still get CUDA_VISIBLE_DEVICES=0; python -c 'import torch; torch.cuda.is_available()' /home/ansible/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:181: UserWarning: CUDA initialization: CUDA...
discuss.pytorch.org