Ubuntu下CUDA8.0和CUDA9.0共存怎么實現

Ubuntu下CUDA8.0和CUDA9.0共存怎么實現

在深度學習和高性能計算領域，CUDA（Compute Unified Device Architecture）是NVIDIA推出的并行計算平臺和編程模型。它允許開發者使用NVIDIA GPU進行通用計算，從而加速計算密集型任務。然而，不同的深度學習框架或應用程序可能需要不同版本的CUDA庫。因此，在某些情況下，用戶可能需要在同一臺機器上安裝多個CUDA版本，并在它們之間進行切換。

本文將詳細介紹如何在Ubuntu系統上實現CUDA 8.0和CUDA 9.0的共存，并展示如何在不同版本之間進行切換。

1. 準備工作

在開始之前，請確保你已經安裝了NVIDIA驅動程序和CUDA工具包。如果你還沒有安裝，可以參考NVIDIA官方文檔進行安裝。

1.1 檢查NVIDIA驅動

首先，檢查你的系統是否已經安裝了NVIDIA驅動程序：

nvidia-smi

如果命令輸出顯示了GPU的信息，說明NVIDIA驅動已經安裝成功。

1.2 下載CUDA安裝包

從NVIDIA官方網站下載CUDA 8.0和CUDA 9.0的安裝包。你可以選擇下載.run文件，這樣可以更靈活地管理多個CUDA版本。

• CUDA 8.0下載頁面
• CUDA 9.0下載頁面

2. 安裝CUDA 8.0和CUDA 9.0

2.1 安裝CUDA 8.0

1. 首先，給下載的CUDA 8.0安裝包賦予執行權限：

   chmod +x cuda_8.0.61_375.26_linux.run

1. 運行安裝程序：

   sudo ./cuda_8.0.61_375.26_linux.run

1. 在安裝過程中，系統會提示你選擇安裝選項。確保不要安裝NVIDIA驅動程序（因為我們已經安裝了最新的驅動程序），并且不要安裝CUDA示例（除非你需要它們）。你可以選擇安裝CUDA工具包和CUDA示例。

2. 安裝完成后，CUDA 8.0將被安裝到/usr/local/cuda-8.0目錄下。

2.2 安裝CUDA 9.0

1. 同樣，給下載的CUDA 9.0安裝包賦予執行權限：

   chmod +x cuda_9.0.176_384.81_linux.run

1. 運行安裝程序：

   sudo ./cuda_9.0.176_384.81_linux.run

1. 在安裝過程中，同樣確保不要安裝NVIDIA驅動程序，并且不要安裝CUDA示例。

2. 安裝完成后，CUDA 9.0將被安裝到/usr/local/cuda-9.0目錄下。

3. 配置環境變量

為了實現CUDA 8.0和CUDA 9.0的共存，我們需要通過環境變量來切換不同的CUDA版本。

3.1 編輯.bashrc文件

打開你的.bashrc文件：

nano ~/.bashrc

在文件末尾添加以下內容：

# CUDA 8.0
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

# CUDA 9.0
# export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
# export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3.2 切換CUDA版本

默認情況下，我們使用CUDA 8.0。如果你需要切換到CUDA 9.0，只需將.bashrc文件中的CUDA 8.0的路徑注釋掉，并取消注釋CUDA 9.0的路徑：

# CUDA 8.0
# export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
# export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

# CUDA 9.0
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存并關閉文件，然后運行以下命令使更改生效：

source ~/.bashrc

3.3 驗證CUDA版本

你可以通過以下命令驗證當前使用的CUDA版本：

nvcc --version

如果輸出顯示的是CUDA 8.0或CUDA 9.0，說明切換成功。

4. 使用符號鏈接管理CUDA版本

除了手動編輯.bashrc文件外，你還可以使用符號鏈接來管理CUDA版本。這種方法更加靈活，適合需要頻繁切換CUDA版本的用戶。

4.1 創建符號鏈接

首先，刪除現有的/usr/local/cuda符號鏈接：

sudo rm -rf /usr/local/cuda

然后，創建一個指向CUDA 8.0的符號鏈接：

sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda

或者，創建一個指向CUDA 9.0的符號鏈接：

sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda

4.2 驗證符號鏈接

你可以通過以下命令驗證符號鏈接是否指向正確的CUDA版本：

ls -l /usr/local/cuda

如果輸出顯示符號鏈接指向/usr/local/cuda-8.0或/usr/local/cuda-9.0，說明符號鏈接創建成功。

5. 總結

通過以上步驟，你可以在Ubuntu系統上實現CUDA 8.0和CUDA 9.0的共存，并且可以根據需要輕松切換不同的CUDA版本。無論是通過環境變量還是符號鏈接，你都可以靈活地管理多個CUDA版本，以滿足不同應用程序的需求。

希望本文對你有所幫助！如果你有任何問題或建議，歡迎在評論區留言。