报错如下:
No CUDA GPUs are available
解决方法:
1、首先在报错的位置net.cuda
前加入cuda检测语句:
print(torch.cuda.is_available())
输出为False
,证明cuda不可用
2、检查本机中的cuda是否安装成功,且版本号是否与pytorch的版本号对应。检查发现没有问题
3、检查os.environ["CUDA_VISIBLE_DEVICES"] = "1"
语句,将1改为0,再运行无误。原因可能是本机上只有一个cuda,序号应为0。
更多推荐
日常Debug——No CUDA GPUs are available
发布评论