一、準備資源

1.cuda9.0.176

2.下載pytorch0.4.0

二、開始安裝

1.先安裝cuda9.0

2.安裝pytorch0.4.0

conda install mkl==2019.1

conda install six==1.12.0

conda install mkl_random==1.0.2

conda install numpy==1.15.4

conda install cffi==1.11.5

conda install freetype==2.9.1

conda install jpeg

conda install libtiff==4.0.9

conda install ninja==1.8.2

conda install olefile==0.46

conda install pillow==5.3.0

conda install cudatoolkit==9.0

conda install cudnn

conda install pytorch-0.4.0-py36_cuda9.0.176_cudnn7.1.2_1.tar.bz2

conda install torchvision

三、驗證

python

import torch torchvision