一、安装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