问题解决：pytorch版本与cuda版本不匹配

知乎

cat /usr/local/cuda/version.txt
# 显示结果：CUDA Version 10.0.130

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

# CUDA 10.0
pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html