问题:在跑深度神经网络时,出现“RuntimeError: No CUDA GPUs are available
”
但是可以明确,本电脑上的GPU是可用的,通过一下方式可以检验:
watch nvidia-smi
import torch
torch.cuda.device_count()
原因:
我的电脑上只有一个GPU,但是代码中有这样的一句话:os.environ['CUDA_VISIBLE_DEVICES'] = '4'
,使得本来正常的GPU环境,突然就变得不可用了
解决方法:
将os.environ['CUDA_VISIBLE_DEVICES'] = '4'
注释掉或者将4改为0