创建python3.6的环境
conda create -n python36 pip python=3.6
激活环境
conda activate python36
添加镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
退cuda版本,同时降低pytorch版本,这里给出方案是,使用cuda10.0和pytorch1.2.0
conda install pytorch torchvision cudatoolkit=10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
最后别忘记,测试一下
import torch
data=torch.Tensor(5,3)
data.cuda()