1. 程式人生 > 其它 >Linux 安裝多版本CUDA

Linux 安裝多版本CUDA

同樣的在執行不同程式碼的時候,需要的環境不同,所以需要不同的GPU環境,Cuda的需要自然也就發生改變 特別是關於tensorflow的執行中。

首先我們需要在nvidia的官網上找到我們所需要的cuda版本,接下來可以根據各自的實際要求來選擇

當前環境配置:Ubuntu20.04 gcc-5, gcc-8, gcc-9, cuda 11.3

需求安裝cuda 9

然後執行安裝這個runfile sudo sh xxx.run

注意不要安裝英偉達驅動 因為之前已經有一個cuda環境 安裝過了

然後修改cuda的環境變數 vim ~/.bashrc (如果環境變數使用的是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

source ~/.bashrc儲存環境變數,使配置生效

進入cd /usr/local

發現cuda其實一個軟連線,所以以後我們需要切換cuda版本的時候,只需要修改這個軟連線的指向就可以了

sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.5 /usr/local/cuda