下载pytorch对应的版本的地址:
http://download.pytorch.org/whl/cu90/torch-1.0.0-cp35-cp35m-linux_x86_64.whl
http://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-linux_x86_64.whl
http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-linux_x86_64.whl
http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl
1,创建虚拟环境
conda create -n pytorch_py3 python=3.6
source activate pytorch_py3
conda install numpy
conda install matplotlib
2,安装pytorch
linux - conda - python 3.6 - CUDA9.0
conda install pytorch torchvision -c pytorch
3,查看cuda与cudnn的版本信息
nvcc --version
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2