Ubuntu16.04下安裝切換多版本cuda和cudnn

2021 年 1 月 22 日
AI
深度學習

前言

實驗室服務器基於Ubuntu16.04平台，多人共用，就難免出現大家對代碼環境要求的不一致，為了解決這一衝突，往往在服務器上會安裝多個版本的cuda。所以此篇博文主要記錄下裝配過程。

安裝cuda

安裝cuda之前，首先要確定電腦中是否已有顯卡驅動，以及驅動的版本（版本會決定這台電腦最高支持cuda的版本），這裡給一下歷史版本的cuda鏈接。

根據電腦平台以及版本的不同，選擇不同的安裝包進行下載（如下）；

安裝按照官方的指引即可；以runfile為例：

Installation Instructions:
Run `sudo sh cuda_10.1.105_418.39_linux.run
Follow the command-line prompts

過程中會出現一些選項，最主要的就是問你要不要安裝顯卡驅動，一定要選no（只要不是新電腦，大多數電腦中是已經有了對應的驅動）。安裝好後會發現在/usr/local/下看到安裝好的cuda-10.1文件夾。

安裝cudnn

同樣我們去nvidia官網下載cudnn。這一步需要註冊nvidia的賬號，選擇對應的版本。

以我們選擇的Download cuDNN v8.0.5 (November 9th, 2020), for CUDA 10.1為例，下載好文件格式是.solitairetheme8，想要解壓的話需要先轉成tgz格式再解壓：

cp  cudnn-10.1-linux-x64-v5.1.solitairetheme8 cudnn-10.1-linux-x64-v8.0.5.tgz
tar -xvf cudnn-10.1-linux-x64-v8.0.5.tgz

命令行解壓然後複製lib64和include文件夾到usr/local/cuda-10.1即可，命令如下：

sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64 
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*#### 多版本cuda的切換

多版本cuda的切換

命令行vim ~/.bashrc打開.bashrc，添加如下行

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

到這一步我們已經設置好了鏈接庫，最後我們只需要在終端輸入

sudo rm -rf /usr/local/cuda #刪除之前創建的軟鏈接
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda #創建新 cuda 的軟鏈接此時系統的cuda版本就會由之前的版本切換為10.1了

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Apr_24_19:10:27_PDT_2019
Cuda compilation tools, release 10.1, V10.1.168

Tags: 深度學習