Hi @kogli2000
To bypass the issue for now, you can try setting CUDA_VISIBLE_DEVICES to just one of your GPUs when running your script so you do not have to remove your other GPU for it to work.
Hi @kogli2000
To bypass the issue for now, you can try setting CUDA_VISIBLE_DEVICES to just one of your GPUs when running your script so you do not have to remove your other GPU for it to work.