Ubuntu16.04+cuda8.0+cudnn6.0+tensorflow1.3

kekukele發表於2018-03-15

安裝時參考了https://www.jianshu.com/p/69a10d0a24b9

但由於版本不同,這裡主要記錄安裝包路徑, 一些注意事項,和基本過程。

 

CUDA8.0下載地址:https://developer.nvidia.com/cuda-80-ga2-download-archive

選擇runfile型別

開啟下載目錄,然後:

1.sudo sh cuda_8.0.61_375.26_linux.run

注意:在安裝時

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
(y)es/(n)o/(q)uit: n
即這一項不安裝。

2.配置系統環境

輸入$ sudo gedit /etc/profile,開啟 gedit ,最後兩行輸入

export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

3.安裝cudnn6.0

安裝包的地址在https://developer.nvidia.com/rdp/cudnn-download
這裡選擇:uDNN v6.0 Library for Linux
解壓後放在home資料夾下。
然後執行命令
$ cd ~ $ sudo tar xvf cudnn-8.0-linux-x64-v5.1.tgz $ cd cuda/include $ sudo cp *.h /usr/local/include/ $ cd ../lib64 $ sudo cp lib* /usr/local/lib/ $ cd /usr/local/lib# sudo chmod +r libcudnn.so.5.1.5 $ sudo ln -sf libcudnn.so.5.1.5 libcudnn.so.5 $ sudo ln -sf libcudnn.so.5 libcudnn.so $ sudo ldconfig

作者:zormin
連結:https://www.jianshu.com/p/69a10d0a24b9
來源:簡書
著作權歸作者所有。商業轉載請聯絡作者獲得授權,非商業轉載請註明出處。

$ cd ~ $ sudo tar xvf cudnn-8.0-linux-x64-v5.1.tgz $ cd cuda/include $ sudo cp *.h /usr/local/include/ $ cd ../lib64 $ sudo cp lib* /usr/local/lib/ $ cd /usr/local/lib# sudo chmod +r libcudnn.so.5.1.5 $ sudo ln -sf libcudnn.so.5.1.5 libcudnn.so.5 $ sudo ln -sf libcudnn.so.5 libcudnn.so $ sudo ldconfig

作者:zormin
連結:https://www.jianshu.com/p/69a10d0a24b9
來源:簡書
著作權歸作者所有。商業轉載請聯絡作者獲得授權,非商業轉載請註明出處。

$ cd ~
$ sudo tar xvf cudnn-8.0-linux-x64-v6.0.tgz
$ cd cuda/include
$ sudo cp cudnn.h /usr/local/include/

$ cd ../lib64
$ sudo cp lib* /usr/local/lib/
$ cd /usr/local/lib# sudo chmod +r libcudnn.so.6.0.21

$ sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
$ sudo ln -sf libcudnn.so.6 libcudnn.so
$ sudo ldconfig

 

4.安裝tensorflow

pip3 install tensorflow_gpu==1.3