anaconda安装pytorch总是安装cpu版本

---

环境

python3.7 + CUDA10.0 + anaconda4.13.0 + pytorch1.2.0

前言

安装好anaconda后，

conda install pytorch torchvision cudatoolkit=10.0

然后安装的pytorch和torchvision总是cpu版本的，就很无语

做法

打开anaconda prompt

建议不要在base环境下，新建一个conda环境

conda create -n pytorch(这个就是环境名，自己起) python=3.7.0

然后

conda install pytorch torchvision cudatoolkit=10.0

这个时候如果安装的是cpu版本，往下看
1、直接y，安装这个cpu版本的
2、conda uninstall numpy，执行这一句，然后就会发现会自己安装pytorch cuda版本的了，别问为什么，问了我也不知道…
3、最后import numpy，import torch，torch.cuda.is_available()，看看有没有问题
4、https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/torchvision-0.4.0-py37_cu100.tar.bz2 去这个网站下载torchvision，假设下载到了D:\DESKTOP（记下你的下载路径）
5、在anaconda prompt里，cd到你的路径，然后conda install --offline xxx。即
就可以了，我的到这就好啦！其他的matlibplot，pillow直接pip install matlibplot pillow就好