今天爱分享给大家带来RuntimeError: No CUDA GPUs are available【解决办法】,希望能够帮助到大家。
RuntimeError: No CUDA GPUs are available
标题 问题阐述
在使用cuda进行模型训练的时候出现了这样一个错误:
显示说没有可用的GPU,当时我就炸了,我GeForce RTX 2080 Ti的GPU不能用?
问题解决
然后我用代码查看了一下GPU的设备数量:
print(torch.cuda.device_count())
结果显示
到这我就开始怀疑GPU到底能不能用了
然后我就查看了一下GPU能不能使用
1564
print(torch.cuda.is_available())
结果最神奇的事情发生了
居然检测到了可用的GPU!
原来在使用Cuda之前都需要检测一下cuda是否可用,没检测就默认没有,最终问题解决了