本文解說如何在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應會自動變黑名單。接著重開機。
|
|
如果你常常換核心開機,那裝DKMS版的Nvidia驅動會更為妥當:
|
|
建議安裝純文字工具"nvtop"用於查看顯示卡使用率
|
|
安裝後使用nvidia-smi
指令檢查安裝狀況。Nvidia驅動應是開箱即用,無需手動調整X伺服器設定。
有Nvidia顯示卡的筆電安裝顯示卡驅動的方式跟桌上型電腦一樣。不過,Intel+Nvidia的雙GPU筆電,即使裝了Nvidia驅動也可能繼續用Intel的GPU渲染3D,導致3D性能低下。
這篇文章有討論過PRIME的用法,或者用Optimius Manager完全切換GPU渲染工作。
另外,如果使用Nvidia專有驅動導致tty黑螢幕的話,於GRUB開機引數加入nomodeset
。
2. 安裝CUDA
CUDA (Compute Unified Device Architecture),在做深度學習、Blender渲染時常用到的技術。
本文撰寫時點CUDA最新版本為11.8。
安裝CUDA,重開機:
|
|
用nvcc --version
指令確認CUDA版本。Arch Linux會將CUDA相關檔案安裝至/opt/cuda
。
欲安裝舊版本CUDA,去AUR找社群打包的就可以了。
例如現在最新版是CUDA 11.8,而你需要CUDA 10,就從AUR裝。注意CUDA最好不要多個版本共存。
|
|
3. 安裝cuDNN
CUDA Deep Neural Network (cuDNN),用於加速神經網路的函式庫。
安裝cuDNN:
|
|
同CUDA,AUR上也有舊版本cuDNN可以安裝。
4. 安裝TensorRT
TensorRT是Nvidia推出的深度學習推理平台。
- 因為軟體授權限制,無法直接用yay裝,需要手動複製AUR儲存庫
|
|
然後到Nvidia官網註冊帳號,下載
tar.gz
檔(注意版本需跟CUDA一致),放到tensorrt
目錄安裝python-oonx
|
|
- 同步依賴套件,編譯安裝
|
|