ubuntu多個cuda與cudnn版本切換

2021-10-18 23:04:49 字數 2888 閱讀 4502

因為要用到tf2.2,需要用到cuda10.1,所以加裝cuda10.1與cudnn7.6.5,並且可以隨時切換舊版本cuda

sudo sh cuda_10.1.105_418.39_linux.run

accept

取消安裝driver

install

a symlink already exists at /usr/local/cuda. update to this installation?

選擇no,否則會建立乙個軟連線覆蓋之前那個/usr/local/cuda

檢視環境變數:

sudo gedit ~/bashrc

原來我配的環境變數是:

export ld_library_path=ldl

ibra

rypa

th:/

usr/

loca

l/cu

da

−10.0/l

ib64e

xpor

tpat

h=

ld_library_path:/usr/local/cuda-10.0/lib64 export path=

ldl​ib

rary

p​at

h:/u

sr/l

ocal

/cud

a−10

.0/l

ib64

expo

rtpa

th=path:/usr/local/cuda-10.0/bin

export cuda_home=$cuda_home:/usr/local/cuda-10.0

但現在因為兩個版本的cuda需要切換,通過建立乙個軟連線/usr/local/cuda指向各自的cuda,所以環境變數改為:

export ld_library_path=ldl

ibra

rypa

th:/

usr/

loca

l/cu

da/l

ib64e

xpor

tpat

h=

ld_library_path:/usr/local/cuda/lib64 export path=

ldl​ib

rary

p​at

h:/u

sr/l

ocal

/cud

a/li

b64e

xpor

tpat

h=path:/usr/local/cuda/bin

export cuda_home=$cuda_home:/usr/local/cuda

source ~/.bashrc 儲存操作

安裝cudnn7.6.5

準備好cuda相對應的cudnn檔案,將檔案解壓得到cuda資料夾,在終端中開啟cuda資料夾的位置輸入以下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include/

sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64/

sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h

sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

建立/usr/local/cuda軟連線指向/usr/local/cuda10.0或/usr/local/cuda10.1,具體根據自己想要的版本切換:

cd /usr/local #務必記住建立軟連線一定要進到該軟連線想要在的資料夾路徑

sudo rm -r cuda

cuda10.0:

sudo ln -s cuda-10.0 cuda # 更改為cuda10.0

sudo ln -s cuda-10.1 cuda # 更改為cuda10.1

也可寫.sh來一鍵切換:

#! /bin/bash

# employ bash shell

if[ -d "/usr/local/cuda"];

then

echo

"cuda資料夾存在"

sudo

rm -r /usr/local/cuda

else

echo

"將重新建立軟連線。"

fisudo

ln -s /usr/local/cuda-10.0 /usr/local/cuda

echo

"現將cuda更換為:"

nvcc -v

echo

"現將cudnn更換為:"

cat /usr/local/cuda/include/cudnn.h |

grep cudnn_major -a 2

#! /bin/bash

# employ bash shell

if[ -d "/usr/local/cuda"];

then

echo

"cuda資料夾存在"

sudo

rm -r /usr/local/cuda

else

echo

"將重新建立軟連線。"

fisudo

ln -s /usr/local/cuda-10.1 /usr/local/cuda

echo

"現將cuda更換為:"

nvcc -v

echo

"現將cudnn更換為:"

cat /usr/local/cuda/include/cudnn.h |

grep cudnn_major -a 2

Ubuntu16 04下安裝CUDA和cuDNN

如果有舊的nvidia驅動,需要先解除安裝舊的驅動 現在有驅動先不安了,以後再更新 cd downloads安裝cuda sudo sh cuda 8.0.44 linux.run注意安裝過程中提示是否安裝驅動的時候選擇n,其他都選擇是。新增環境變數到 bashrc export cuda home...

ubuntu下安裝多個cuda和cudnn

參考部落格 大概意思指的是安裝pytorch版本時候編譯的cuda版本是7050,而現在使用的是cuda8.0以上的新版本在呼叫pytorch,所以造成不匹配.可能原因 每乙個pytorch版本都有對應的cuda版本,可能是在安裝pytorch的時候,選擇的pytorch版本所對應的版本cuda版本...

ubuntu 安裝多個CUDA版本並可以隨時切換

cuda是什麼就不介紹了,直接講怎麼實現cuda多版本的共存和實時切換。1 安裝多個版本的cuda 這裡,我們以cuda9 1版本和cuda9 0版本為例 先安裝哪個無所謂 首先,在cuda版本庫中選擇自己需要的cuda版本。然後,選擇對應的安裝包,這裡選擇runfile型別的安裝檔案,以便後面設定...