1. 程式人生 > >安裝GPU版本的tensorflow過程總結

安裝GPU版本的tensorflow過程總結

以及 () tor source api versions orf 系統 linux

首先安裝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模式下的tesorflowcnn時,出現

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

然後將這個壓縮包解壓到主文件夾,會自動命名為cudacuda文件夾下有includelib64兩個文件夾

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/ #這裏*是通配符,

lib*指的是名字中帶有lib的所有文件

3、在系統路徑下建立軟鏈接(解壓出來的lib64下面有3so文件。分別是 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過程總結