教你在Arch Linux安裝Nvidia、CUDA、cuDNN、TensorRT專有驅動程式

本文解說如何在Arch Linux安裝Nvidia專有驅動，以及相關的CUDA、cuDNN、TensorRT套件。

儘管Nvidia官方有詳細的驅動安裝文件，但壓根沒提到Arch Linux系的發行版。所幸Arch Linux還是可以安裝Nvidia推出的各式技術與產品的，多虧了全球社群的協助。

一般用戶玩遊戲只要裝完Nvidia專有驅動即可。要跑3D渲染、神經網路(Neural Network)、深度學習(Deep Learning)等AI相關程式的用戶才需要裝後面的CUDA、cuDNN、TensorRT套件。

另請參閱此篇安裝Anaconda，可輕鬆解決Python版本問題。

1. 安裝Nvidia專有驅動

Arch Linux安裝Nvidia驅動只需要一行指令，安裝後nouveau應會自動變黑名單。接著重開機。

1
sudo pacman -S nvidia nvidia-utils

如果你常常換核心開機，那裝DKMS版的Nvidia驅動會更為妥當：

1
sudo pacman -S nvidia-dkms nvidia-utils

建議安裝純文字工具"nvtop"用於查看顯示卡使用率

1
sudo pacman -S nvtop

安裝後使用nvidia-smi指令檢查安裝狀況。Nvidia驅動應是開箱即用，無需手動調整X伺服器設定。

有Nvidia顯示卡的筆電安裝顯示卡驅動的方式跟桌上型電腦一樣。不過，Intel＋Nvidia的雙GPU筆電，即使裝了Nvidia驅動也可能繼續用Intel的GPU渲染3D，導致3D性能低下。

這篇文章有討論過PRIME的用法，或者用Optimius Manager完全切換GPU渲染工作。

另外，如果使用Nvidia專有驅動導致tty黑螢幕的話，於GRUB開機引數加入nomodeset。

CUDA (Compute Unified Device Architecture)，在做深度學習、Blender渲染時常用到的技術。

本文撰寫時點CUDA最新版本為11.8。

安裝CUDA，重開機：

1
sudo pacman -S cuda

用nvcc --version指令確認CUDA版本。Arch Linux會將CUDA相關檔案安裝至/opt/cuda。

欲安裝舊版本CUDA，去AUR找社群打包的就可以了。

例如現在最新版是CUDA 11.8，而你需要CUDA 10，就從AUR裝。注意CUDA最好不要多個版本共存。

1
yay -S cuda-10.0

CUDA Deep Neural Network (cuDNN)，用於加速神經網路的函式庫。

安裝cuDNN：

1
sudo pacman -S cudnn

同CUDA，AUR上也有舊版本cuDNN可以安裝。

TensorRT是Nvidia推出的深度學習推理平台。

1
2
sudo pacman -S base-devel
git clone https://aur.archlinux.org/tensorrt.git

1
yay -S python-onnx python-onnxruntime python-tensorflow-cuda

1
2
3
cd tensorrt
makepkg --syncdeps
makepkg --install