背景
import torch 没问题
但print(torch.cuda.is_available())总是FALSE
CUDA和cudnn因为tensorflow调用成功肯定是没问题的,所以不再检查这两个。
方法
检查:
1.cudatoolkit是否安装
2.检查pytorch的version装成只有cpu版的了?(cpu版显示)
避免这种事情的最简单方法:
使用官网连接直接conda安装pytorch,cudatoolkit安装慢的话用conda提供的url下载好,cd到下载的目录使用conda install --use-local 加文件名,安好再用一遍官网连接确定连接包配置正确即可。