ubuntu安裝並管理多版本的cuda

1. 下載cuda

各個版本的CUDA :https://developer.nvidia.com/cuda-toolkit-archive

2. 下載完cuda後,使用命令解壓

sudo sh cuda_9.0.176_384.81_linux.run

3. 安裝過程
執行完上述命令後,是一堆協議說明
直接按q退出協議說明.
然後需要輸入

accept/decline/quit: accept  #接受協議
 Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26? 
  y)es/(n)o/(q)uit: n  #是否顯卡驅動包,由於已經安裝顯卡驅動,選擇n
  
  Install the CUDA 9.0 Toolkit?
  (y)es/(n)o/(q)uit: y #是否安裝工具包,選擇y
 
 Enter Toolkit Location
 [ default is /usr/local/cuda-9.0 ]: #工具包安裝地址,默認回車即可
 
 Do you want to install a symbolic link at /usr/local/cuda?
 (y)es/(n)o/(q)uit: n #添加鏈接**注意這個連接,如果你之前安裝過另一個版本的cuda,
              #除非你確定想要用這個新版本的cuda,否則這裏就建議選no,因爲指定該鏈接後會將cuda指向這個新的版本**
 
 Install the CUDA 9.0 Samples?
 (y)es/(n)o/(q)uit: y #安裝樣例
 
 Enter CUDA Samples Location
  [ default is /root ]:  #樣例安裝地址默認即可

4. 執行命令

sudo gedit ~/.bashrc 打開.bashrc文件

5. 在.bashrc文件內部 添加如下指令:

如果你之前已經有了cuda版本,如下:

 export PATH=$PATH:/usr/local/cuda-8.0/bin
 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64/
 export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-8.0/lib64

只需將cuda-8.0換成cuda 如下:

 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
 export PATH=$PATH:/usr/local/cuda/bin
 export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

一定要記得保存文件,一定要記得保存文件,一定要記得保存文件。

6. 執行命令,激活環境

1 source ~/.bashrc

7. cuda版本切換,刪除原來的軟連接

sudo rm -rf /usr/local/cuda

8. 建立新的軟連接

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

9. 驗證切換是否成功

多個cuda版本,可以在/usr/local/目錄下查看自己安裝的cuda版本,命令:

 cd /usr/local/

同時,在該路徑下執行如下命令

 stat cuda
 nvcc --version
 cat /usr/local/cuda/version.txt

10. 如果執行完 第2 、3條命令後,顯示的都是同一版本的cuda,那麼則成功,否則,不一致則跳轉11

11. 解決方案

執行如下命令:

sudo gedit ~/.bashrc 

打開.bashrc文件 ,把之前步驟5的三行指令換成 如下指令:

 export PATH=/usr/local/cuda/bin:$PATH
 export LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATH
 export CUDA_HOME=/usr/local/cuda

一定要記得保存文件,一定要記得保存文件,一定要記得保存文件。

執行如下激活指令:

source ~/.bashrc

12. 補充安裝cudnn

tar -zxvf cudnn-9.2-linux-x64-v7.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章