1. 首页
  2. 数据库
  3. 其它
  4. torch.cuda.is_available()返回False解决方案

torch.cuda.is_available()返回False解决方案

上传者: 2020-12-30 20:41:09上传 PDF文件 67.4KB 热度 22次
在配置完环境以后运行程序,程序正常运行但是GPU没有任务,此时判断程序没有在GPU上运行。调试步骤: 先查看cuda安装有没有问题:nvcc -V 可以看到cuda驱动等安装没有问题。 进程序去看.cuda()或者.todevice之类代码,看程序确实放在GPU上了,故排除 但是在查看代码是看到这里是一个and,参数args.no_cuda是设置的Flase,所以问题因为在torch.cuda.is_available(),没有获得cuda加速。 查看深度学习环境搭建torch.cuda.is_available()这里返回了
用户评论