安裝NVIDIA GPU顯示卡驅動

如果需要安裝cuda版本的PyTorch，電腦也有獨立顯示卡的時候，一般需要更新一下Ubuntu獨立顯示卡驅動。否則即使安裝了cuda版本的PyTorch也沒辦法使用GPU。

因為我的膝上型電腦顯示卡是NVIDIA的，所以首先去官網：NVIDIA顯示卡驅動下載，檢視適合自己顯示卡的驅動，下載runfile檔案：NVIDIA-Linux-x86_64-384.98.run。

然後按Ctrl + Alt + F1到控制檯，關閉當前圖形環境：

sudo service lightdm stop

解除安裝可能存在的舊版本NVIDIA 驅動（這一步是必須的，不然重啟會有問題）:

sudo apt-get remove --purge nvidia*

安裝驅動可能需要的依賴：

sudo apt-get update
sudo apt-get install dkms build-essential linux-headers-generic

把 nouveau 驅動加入黑名單並禁用用 nouveau 核心模組（如果新重灌的系統，沒有驅動起作用，忽略）：

sudo nano /etc/modprobe.d/blacklist-nouveau.conf

在檔案 blacklist-nouveau.conf 中加入如下內容：

blacklist nouveau
options nouveau modeset=0

儲存退出，執行：

sudo update-initramfs -u

然後重啟：

reboot

重啟後再次進入字元終端介面（Ctrl + Alt + F1），並關閉圖形介面：

sudo service lightdm stop

進入之前NVIDIA驅動檔案下載目錄，安裝驅動：

sudo chmod u+x NVIDIA-Linux-x86_64-384.98.run
sudo ./NVIDIA-Linux-x86_64-384.98.run -no-opengl-files

-no-opengl-files表示只安裝驅動檔案，不安裝OpenGL檔案。這個引數不可忽略，否則會導致登入介面死迴圈。

最後重新啟動圖形環境：

sudo service lightdm start

可以通過以下命令確認驅動是否正確安裝：

cat /proc/driver/nvidia/version

值得注意的是，安裝完顯示卡驅動後，桌面可能會出現選單工作列及窗體最大化最小化消失問題。解決辦法如下：

首先安裝compiz配置管理器：

sudo apt-get install compizconfig-settings-manager

然後，開啟配置管理器：

sudo ccsm

並在”general”啟用OpenGL，在”Desktop”裡啟用Ubuntu Unity Plugin，然後返回桌面。

接著，重置compiz：

dconf reset -f /org/compiz/

最後登出重啟：

gnome-session-quit

至此，NVIDIA顯示卡驅動安裝成功。

科學上網的簡易方法

進入PyTorch官網：
PyTorch

顯示如下：

許多教程說在“Run this command”下會顯示安裝PyTorch的命令。如果Ubuntu下已經配置過Anaconda的話，可以直接使用conda命令安裝成功PyTorch：

conda install pytorch torchvision cuda80 -c soumith

其中，pytorch是軟體，torchvision是pytorch使用的資料包和庫，cuda80是GPU版本。

但是基於特殊的網路國情，一般無法檢視“Run this command”之後的指令，安裝也不會成功。

下載PyTorch

不能科學上網的話，使用另外一種方法來安裝PyTorch。首先下載所需版本的PyTorch。在PyTorch主頁上點選“Click here for previous versions of PyTorch”，如上圖右下角所示。

找到所需版本，如上圖所示，下載torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl。其中，0.2.0表示PyTorch版本，cp36表示支援Python3.6，cuda80表示支援GPU版本，電腦有GPU的話須選擇該項。

安裝PyTorch

進入PyTorch的下載目錄，使用pip命令安裝：

pip install torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

安裝torchvision

安裝torchvision比較簡單，可直接使用pip命令安裝：

pip install torchvision

至此，pytorch和torchvision已安裝完畢。

更新numpy

安裝成功pytorch和torchvision後，開啟ipython，輸入：

import torch

可能會出現報錯的情況，如下所示：

ImportError: numpy.core.multiarray failed to import

這是因為numpy的版本需要更新，直接使用pip更新numpy：

pip install numpy

至此，PyTorch安裝成功：

最後，驗證GPU是否可用：

返回True，表明已經可以使用GPU了！

參考連結：

我的部落格即將同步至騰訊雲+社群，邀請大家一同入駐。