【环境】
linux ubuntu 16.04 python3.5
【步骤】
1,安装显卡驱动和CUDA10.1:linux ubuntu nvidia 驱动 安全升级 CUDA cudnn 切换_nvcc -v 版本升级-CSDN博客
2,安装torch, torchvision:
# 查看和卸载之前的老版本(如果有)
$ pip3 list
Package Version
----------------------------- -----------
absl-py 0.8.0
apturl 0.5.2
astor 0.8.0
attrs 19.3.0
backcall 0.1.0
bcolz 1.2.1
beautifulsoup4 4.4.1
bleach