CUDA配置好之后 pytorch调用GPU还是失败?

背景

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 加文件名,安好再用一遍官网连接确定连接包配置正确即可。

你可能感兴趣的:(自己踩过的坑,python,人工智能,深度学习)