笔记--Linux环境下无法使用GPU

自己遇到的问题，记个笔记，以备不时之需。pytorch环境下，torch.cuda.is_available()返回False,

1、查看当前pytorch版本，在进行Python后，import torch，用torch.__verison__查看。再查看cuda版本，命令为nvcc -V。

2、在pytorch官网查找对应版本，进行安装。

3、如果官网没有对应的cuda版本，需要转换CUDA版本

然后找到相应命令，进行安装。

 注意：后面的-c pytorch，加上就是用官网的渠道进行下载，一般会很慢，去掉后是根据自己设置的源进行下载。一般推荐去掉-c  pytorch，利用设置的清华源进行下载。