conda安装pytorch
查看cuda 版本 nvcc -V
~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176
cuda版本9.0 对应安装命令
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
#指定pytorch版本
#conda install pytorch=0.4.1 torchvision cudatoolkit=9.0 -c pytorch
使用清华镜像快速安装pytorch
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add chann