ubuntu系统下 安装pytorch, 解决 torch.cuda.is_available() false 的问题

新配置的电脑环境，安装了torch之后，用不了，torch.cuda.is_available()  false。查了好多博客，改来改去，我最后感觉版本都对应上了啊，结果还是不行。后来按照实验室老哥们的方法走了一遭，这也太简单了。我一下午都在改，install uninstall 。以后不行的话就别纠结了，直接重新建一个环境，来一遍吧，贼快。

1.新建一环境：放弃以前那个改的乱七八糟的环境吧，新建一个超快

conda create --name xxx python==3.6

2.给conda加新的镜像：注意不是之前清华的conda源，而是pytorch的。要不然安装的torch版本很低

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

3.查看自己的显卡驱动

nvidia-smi

比如我的如下：

 

4.根据自己的驱动，到pytorch官网上找自己电脑配置对应的conda命令

https://pytorch.org/

就是这个命令 后面的-c pytorch去掉

直接安装就行！！！！

5.安装

conda install pytorch torchvision cudatoolkit=10.1

6.搞定

import torch
torch.cuda.is_available()

得到true，哭了。原来这么简单汗！