RuntimeError: No CUDA GPUs are available【解决办法总结】
- 检查cuda,cudnn,显卡驱动版本是否能对应上。
windows中cmd终端输入nvcc --version可以查看版本。
- 检查cuda是否能够正常使用
import torch
import torchvision
print(torch.cuda.is_available())
a = torch.Tensor(5,3)
a=a.cuda()
print(a)
- 查看配置文件中的id号是否能对应上。 [外链图片转存中…(img-o37VQ80o-1607913861613)]
更多推荐
RuntimeError: No CUDA GPUs are available【解决办法总结】
发布评论