- 环境
- 前言
- 做法
环境
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就好
