之前用pytorch四康,所以已經(jīng)配置好了cuda9.0和cudnn7.0幔嫂,但是在pip3 install tensorflow-gpu安裝成功后,測(cè)試tensorflow的例子是報(bào)錯(cuò)了,兩個(gè)錯(cuò)誤先后報(bào)出:
ImportError:libcudart.so.8.0: cannot open shared object file: No such fileor directory
ImportError:libcudnn.6.0: cannot open shared object file: No such file ordirectory
查了一下原因是因?yàn)閠ensorflow-gpu1.4.1的版本默認(rèn)支持cuda8.0和cudnn6.0吨悍,也不知道如何修改tensorflow的配置,所以只能再安裝cuda8.0和cudnn6.0了
安裝cuda8.0(按照官方的安裝說明即可)
注意這里下載的是cuda8.0的runfile(local)文件蹋嵌。下載完cuda8.0后,執(zhí)行如下語(yǔ)句葫隙,運(yùn)行runfile文件:
sudo sh cuda_8.0.27_linux.run
因?yàn)轵?qū)動(dòng)之前已經(jīng)安裝栽烂,這里就不要選擇安裝驅(qū)動(dòng)。其余的都直接默認(rèn)或者選擇是即可恋脚。 使用:
sudo gedit /etc/profile
打開“profile”文件腺办,在末尾處添加(注意不要有空格,不然會(huì)報(bào)錯(cuò)):
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH
安裝cudnn6.0
cudnn安裝步驟:
從官網(wǎng)上下載cudnn的安裝包糟描,將安裝包解壓,將此安裝包放在home路徑下即可怀喉,并在當(dāng)前路徑下進(jìn)行解壓,解壓后的文件夾名為cuda。
在終端上編輯如下代碼:
(也就是把include和lib64的文件都分別拷貝到/usr/local/cuda/include/和/usr/local/cuda/lib64/中)
cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include/
cd cuda/lib64
sudo cp libcudnn* /usr/local/cuda/lib64/
再運(yùn)行tensorflow的例子船响,就成功了躬拢。