新配置的电脑环境,安装了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命令
就是这个命令 后面的-c pytorch去掉
直接安装就行!!!!
5.安装
conda install pytorch torchvision cudatoolkit=10.1
6.搞定
import torch
torch.cuda.is_available()
得到true,哭了。原来这么简单汗!