Linux 安裝多版本CUDA
阿新 • • 發佈:2022-04-20
同樣的在執行不同程式碼的時候,需要的環境不同,所以需要不同的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