服务器:Ubuntu 16.04
CUDA:10.0

创建一个conda环境：
conda create -n haha python==3.6

安装torch

pip install torch-1.4.0+cu100-cp36-cp36m-linux_x86_64.whl

【https://download.pytorch.org/whl/cu100/torch_stable.html】

安装torchvision

pip install torchvision-0.5.0+cu100-cp36