Win10x64+PyTorch1.2.0

1、安装新版本的anaconda

 

 

2、换为国内源

1. # 配置国内源，方便安装Numpy,Matplotlib等
2.  
   conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
3.  
   conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
4.  
   # 配置国内源，安装PyTorch用
5.  
   conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
6.  
   # 显示源地址
7.  
   conda config --set show_channel_urls yes

3、安装pytorch 

conda install pytorch==1.2.0 cudatoolkit=10.0  #无需手动安装cudnn，会在anaconda中自动下载安装。

实际直接就安装torchvison，会根据pytorch的版本自动安装对应的版本。

pip install torchvision==0.3.6 

测试是否安装成功

 python

import torch

torch.cuda.is_available()

TRUE成功

 

conda remove -n pytorch-gpu --all

conda clean --packages --tarballs

#再重新执行pytorch安装命令，有时网络原因出错，多尝试~

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0