Linux下安裝pytorch的GPU版本

在計算集群提交任務時使用到了GPU，提示如下錯誤：

The NVIDIA driver on your system is too old (found version 9000).
Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.

意思是說pytorch版本和cuda的驅動未對應上，解決方案一是升級驅動，二是更換pytorch版本，可行的只能是二了。

查詢cuda版本

cat /usr/local/cuda/version.txt

輸出結果如下：

CUDA Version 9.1.85
CUDA Patch Version 9.1.85.1
CUDA Patch Version 9.1.85.2
CUDA Patch Version 9.1.85.3

安裝pytorch

在https://pytorch.org/get-started/previous-versions/找到與cuda 9.1對應的pytorch號。雖然沒有9.1版本的，但是9.0的也是兼容的。找到如下結果,為了方便把其他版本的也一並粘貼過來了。

# CUDA 9.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

# CUDA 10.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch

那么，我們需要下載三個指定的包，分別是：pytorch==1.1.0, torchvision==0.3.0和 cudatoolkit=9.0。去anaconda倉庫分別找到這三個東西，下載好，然后分別安裝即可。

conda install cudatoolkit-9.0-h13b8566_0.tar.bz2 
conda install pytorch-1.1.0-py3.7_cuda9.0.176_cudnn7.5.1_0.tar.bz2 
conda install torchvision-0.3.0-py37_cu9.0.176_1.tar.bz2

測試

import torch
torch.__version__

# '1.1.0'