安裝GPU版本的tensorflow過程總結
首先安裝cuda,和cudnn
安裝完了後再用pip去安裝tensorflow的指令如下:$ sudo pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl
在spyder環境下,利用GPU模式下的tesorflow跑cnn時,出現
E tensorflow/stream_executor/cuda/cuda_dnn.cc:390]Loaded runtime CuDNN library: 5005 (compatibility version 5000) but source wascompiled with 5110 (compatibility version 5100). If using a binary install, upgrade your CuDNNlibrary to match. If building fromsources, make sure the library loaded at runtime matches a compatible versionspecified during compile configuration.
F tensorflow/core/kernels/conv_ops.cc:605]Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms)
這個問題
原因:大概就是cudnn版本不符合,我的是5.0,官網要求的是5.1
更新cudnn首先要下載一個cudnn的壓縮包,可以官網上下,也有網盤
鏈接:http://pan.baidu.com/s/1slPfR8x 密碼:g4fv
然後將這個壓縮包解壓到主文件夾,會自動命名為cuda,cuda文件夾下有include和lib64兩個文件夾
1、刪除原來的cudnn系統路徑下的一些文件
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn* #這裏*是通配符,libcudnn*指的是名字中帶有libcudnn的所有文件
2、安裝剛才解壓的cudnn版本,在終端cd到剛解壓的cuda文件夾,然後繼續輸入下面兩個指令,這兩個指令相當於把解壓後的cuda文件夾下的一些文件拷到系統路徑下面
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/ #這裏*是通配符,
3、在系統路徑下建立軟鏈接(解壓出來的lib64下面有3個so文件。分別是 libcudnn.so 和 libcudnn.so.5以及 libcudnn.so.5.1.3文件。 並且這3個點so文件大小都一樣。其實都是軟連接!libcudnn.so鏈接到libcudnn.so.5,而libcudnn.so.5.又鏈接到libcudnn.so.5.1.3。 真正的文件只有libcudnn.so.5.1.3)
cd /usr/local/cuda/lib64
sudo chmod +r libcudnn.so.5.1.10
sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig
安裝GPU版本的tensorflow過程總結