溫馨提示×

安裝NVLink驅(qū)動在Ubuntu上

小樊
125
2024-09-05 04:36:03
欄目: 智能運維

NVLink是NVIDIA的一種高速GPU互連技術(shù),用于連接多個GPU以實現(xiàn)更高的計算性能

  1. 首先,確保您的系統(tǒng)已安裝了NVIDIA CUDA工具包。如果沒有,請訪問NVIDIA CUDA工具包下載頁面并按照說明進行安裝。

  2. 安裝完成后,重啟計算機以使更改生效。

  3. 打開終端并運行以下命令以安裝NVLink驅(qū)動:

sudo apt update
sudo apt install nvidia-driver-<version>-nvlink

<version>替換為與您的CUDA工具包匹配的NVIDIA驅(qū)動程序版本。例如,如果您安裝了CUDA 11.0,則可以使用以下命令:

sudo apt install nvidia-driver-450-nvlink
  1. 安裝完成后,重啟計算機以使更改生效。

  2. 要驗證NVLink是否正常工作,請運行以下命令:

nvidia-smi nvlink

這將顯示有關(guān)NVLink連接的信息,包括連接狀態(tài)、數(shù)據(jù)傳輸速率等。

現(xiàn)在,您已經(jīng)在Ubuntu上安裝了NVLink驅(qū)動。如果您在安裝過程中遇到任何問題,請查閱NVIDIA官方文檔以獲取更多幫助。

0