【环境】
linux ubuntu 16.04 python3.5
【步骤】
1，安装显卡驱动和CUDA10.1：https://blog.csdn.net/qxqxqzzz/article/details/102794226
2，安装torch, torchvision:
# 查看和卸载之前的老版本（如果有）
$ pip3 listPackage 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 3.1.0
blinker 1.3
Bottleneck 1.2.1
Brlapi 0.6.4
certifi 2019.9.11
chardet 2.3.0
checkbox-support 0.22
command-not-found 0.3
cryptography 1.2.3
cycler 0.10.0
Cython 0.29.13
decorator 4.4.0
defer 1.0.6
defusedxml 0.6.0
descartes 1.1.0
entrypoints 0.3
feather-format 0.4.0
feedparser 5.1.3
gast 0.2.2
google-pasta 0.1.7
graphviz 0.13
grpcio 1.24.1
guacamole 0.9.2
h5py 2.10.0
html5lib 0.999
httplib2 0.9.1
idna 2.0
imageio 2.5.0
imgaug 0.3.0
importlib-metadata 0.23
ipykernel 5.1.3
ipython 7.8.0
ipython-genutils 0.2.0
ipywidgets 7.5.1
isoweek 1.3.3
jedi 0.15.1
Jinja2 2.8
joblib 0.13.2
jsonschema 3.1.1
jupyter 1.0.0
jupyter-client 5.3.4
jupyter-console 6.0.0
jupyter-core 4.6.1
Keras-Applications 1.0.8
Keras-Preprocessing 1.1.0
kiwisolver 1.1.0
language-selector 0.1
louis 2.6.4
lxml 3.5.0
Mako 1.0.3
Markdown 3.1.1
MarkupSafe 0.23
matplotlib 3.0.3
mistune 0.8.4
mizani 0.5.4
more-itertools 7.2.0
nbconvert 5.6.1
nbformat 4.4.0
networkx 2.3
notebook 6.0.1
numpy 1.17.3
oauthlib 1.0.3
onboard 1.2.0
opencv-python 4.1.1.26
opt-einsum 3.1.0
padme 1.1.1
palettable 3.3.0
pandas 0.24.2
pandas-summary 0.0.7
pandocfilters 1.4.2
parso 0.5.1
patsy 0.5.1
pexpect 4.0.1
pickleshare 0.7.5
Pillow 6.2.1
pip 19.2.3
plainbox 0.25
plotnine 0.5.1
progressbar 2.5
progressbar2 3.46.1
prometheus-client 0.7.1
prompt-toolkit 2.0.9
protobuf 3.10.0
ptyprocess 0.5
pyarrow 0.15.0
pyasn1 0.1.9
pycups 1.9.73
pycurl 7.43.0
Pygments 2.4.2
pygobject 3.20.0
PyJWT 1.3.0
pyparsing 2.0.3
pypng 0.0.20
pyrsistent 0.15.5
python-apt 1.1.0b1
python-dateutil 2.8.0
python-debian 0.1.27
python-systemd 231
python-utils 2.3.0
pytz 2014.10
PyWavelets 1.0.3
pyxdg 0.25
PyYAML 5.1.2
pyzmq 18.1.0
qtconsole 4.5.5
reportlab 3.3.0
requests 2.9.1
scikit-image 0.15.0
scikit-learn 0.21.3
scipy 1.3.1
screen-resolution-extra 0.0.0
seaborn 0.9.0
Send2Trash 1.5.0
sessioninstaller 0.0.0
setuptools 39.1.0
Shapely 1.6.4.post2
simplegeneric 0.8.1
six 1.12.0
sklearn 0.0
sklearn-pandas 1.8.0
ssh-import-id 5.5
statsmodels 0.10.1
system-service 0.3
tensorboard 1.10.0
tensorboardX 1.8
tensorflow-estimator 2.0.0
tensorflow-gpu 1.10.0
termcolor 1.1.0
terminado 0.8.2
testpath 0.4.2
torch 0.4.1
torchtext 0.4.0
torchvision 0.2.2.post3
tornado 6.0.3
tqdm 4.36.1
traitlets 4.3.2
ubuntu-drivers-common 0.0.0
ufw 0.35
unattended-upgrades 0.1
unity-scope-calculator 0.1
unity-scope-chromiumbookmarks 0.1
unity-scope-colourlovers 0.1
unity-scope-devhelp 0.1
unity-scope-firefoxbookmarks 0.1
unity-scope-gdrive 0.7
unity-scope-manpages 0.1
unity-scope-openclipart 0.1
unity-scope-texdoc 0.1
unity-scope-tomboy 0.1
unity-scope-virtualbox 0.1
unity-scope-yelp 0.1
unity-scope-zotero 0.1
urllib3 1.13.1
usb-creator 0.3.0
virtualenv 15.0.1
wcwidth 0.1.7
webencodings 0.5.1
Werkzeug 0.16.0
wheel 0.29.0
widgetsnbextension 3.5.1
wrapt 1.11.2
xdiagnose 3.8.4
xkit 0.0.0
XlsxWriter 0.7.3
zipp 0.6.0# 卸载老版本
$ sudo pip3 uninstall torch torchvision
# 安装新版本 （内网环境），按照：https://pytorch.org/get-started/locally/ 指示说明
$ pip3 --trusted-host=pypi.org --trusted-host=files.pythonhosted.org --trusted-host=download.pytorch.org install torch torchvision --user【参考】https://pytorch.org/get-started/locally/