一、安裝pytorch 0.4.1 cuda10
1、先conda創建環境
2、在環境中輸入指令
pip install torch==0.4.1
3、安裝 torchvision
pip install torchvision==0.2.2
torch 0.4.1 版本對應的 torchvision版本爲 0.2.2
二、安裝pytorch 0.4.1 cuda 9
pip3 install torch==0.4.1+cu92 torchvision==0.2.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html