非root 切換 cudnn

在linux系統上使用tensorflow的時候經常會碰到cudnn 版本不匹配的問題

這裏介紹一種簡單易用的方法切換cudnn供參考使用

 

首先去官網或其他可以下載的地方,下載指定版本的cudnn 解壓包

舉例:cudnn-9.0-linux-x64-v7.6.1.34.gtz 是cudnn7.6 版本的,而root權限下是cudnn7.3版

在使用tensorflow2.0時,需要使用cudnn7.6版

那麼下載cudnn-9.0-linux-x64-v7.6.1.34.gtz 這個文件並解壓縮文件能夠得到include 和lib64 文件

這時候在非root的根目錄下的.local(如果沒有可以mkdir .local 創建) 軟連接或者複製剛剛解壓的文件到該目錄下,並在~/.bachrc

裏面添加如下語句:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/用戶名/.local/lib64

然後更新環境就可以使用cudnn7.6

 

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