Closed KatrionaGoldmann closed 1 year ago
Running the models on Baskerville crashes when requesting GPUs, despite torch.is_cuda_available() returning True and the set-up selecting multiple GPUs.
torch.is_cuda_available()
True
Using:
#SBATCH --gpus-per-task 3 #SBATCH --tasks-per-node 1 #SBATCH --nodes 1
Returns error RuntimeError: CUDA error: no kernel image is available for execution on the device
RuntimeError: CUDA error: no kernel image is available for execution on the device
Need to load compatible versions of CUDA, Pytorch and Python on the server.
Currently running:
For more info see:
Running the models on Baskerville crashes when requesting GPUs, despite
torch.is_cuda_available()
returningTrue
and the set-up selecting multiple GPUs.Using:
Returns error
RuntimeError: CUDA error: no kernel image is available for execution on the device