如何在Ubuntu中啟用NVLink

小樊
88
2024-09-05 04:38:43

要在Ubuntu中啟用NVLink,您需要確保您的系統(tǒng)支持它,然后安裝NVIDIA專有驅(qū)動(dòng)程序。以下是在Ubuntu上啟用NVLink的逐步指南:

準(zhǔn)備工作:

  1. 檢查兼容性:確保您的GPU(例如,NVIDIA RTX 30系列)支持NVLink。
  2. 更新系統(tǒng):通過(guò)運(yùn)行以下命令來(lái)更新您的Ubuntu系統(tǒng):
    sudo apt update
    sudo apt upgrade
    
  3. 安裝NVIDIA專有驅(qū)動(dòng)程序:從NVIDIA官方網(wǎng)站下載與您的GPU兼容的最新驅(qū)動(dòng)程序。

啟用NVLink:

  1. 關(guān)閉電腦:為了安全起見,最好在繼續(xù)之前關(guān)閉電腦。
  2. 安裝驅(qū)動(dòng)程序:按照NVIDIA提供的說(shuō)明安裝驅(qū)動(dòng)程序。這通常涉及下載一個(gè).run文件,然后在終端中導(dǎo)航到該文件的位置并運(yùn)行它。例如:
    chmod +x NVIDIA-Linux-x86_64-<version>.run
    sudo ./NVIDIA-Linux-x86_64-<version>.run
    
    按照屏幕上的指示完成安裝,您可能需要選擇“是”以接受許可協(xié)議并啟用驅(qū)動(dòng)程序。
  3. 重啟電腦:安裝完成后重啟電腦。
  4. 檢查NVLink狀態(tài):重啟后,打開終端并運(yùn)行以下命令來(lái)檢查NVLink是否已啟用:
    nvidia-smi
    
    在輸出中,您應(yīng)該能夠看到與NVLink相關(guān)的信息。如果它被標(biāo)記為啟用,那么您就已經(jīng)成功地在Ubuntu中啟用了NVLink。

額外提示:

  • NVLink通常用于連接支持NVLink的GPU,以提供更高的帶寬和更好的性能。
  • 如果您有多個(gè)GPU,并且它們都支持NVLink,您可能需要在BIOS/UEFI設(shè)置中啟用它。
  • 請(qǐng)記住,使用專有驅(qū)動(dòng)程序可能會(huì)使您的系統(tǒng)變得不再符合Ubuntu的官方支持政策。
  • 如果您遇到任何問(wèn)題,可以參考NVIDIA的官方文檔或社區(qū)論壇尋求幫助。

請(qǐng)注意,隨著Linux發(fā)行版和NVIDIA驅(qū)動(dòng)程序版本的更新,上述步驟可能會(huì)有所變化。始終建議查看最新的官方文檔以獲取最準(zhǔn)確的信息。

0