能夠在官網查看通過測試的TensorFlow版本對應的CUDA和Cudnn:https://tensorflow.google.cn/install/source#linuxhtml
Tensorflow-gpu安裝python
1、環境變量
OS:ubuntu-18.04linux
編程語言:Python3.6編程
2、安裝步驟
2.1搞清楚要安裝TensorFlow-GPU版本對應的CUDA和Cudnn,具體以下表:
2.2下載Ubuntu對應版本的驅動,CUDA以及Cudnn(注NVIDIA驅動裝最新版)
NVIDIA驅動(.run)下載地址:https://www.nvidia.cn/Download/index.aspx?lang=cnubuntu
CUDA(.run)歷史版本下載地址:https://developer.nvidia.com/cuda-toolkit-archivevim
cuDNN(.deb)下載地址:https://developer.nvidia.com/rdp/cudnn-archive框架
2.3 Ubuntu下安裝NVIDIA驅動
1)關閉UEFI Secure Boot選項(很重要)編程語言
2)打開終端,刪除舊的驅動:學習
sudo apt-get purge nvidia*測試 |
3)禁用自帶的 nouveau nvidia驅動
sudo vim /etc/modprobe.d/blacklist-nouveau.conf |
在vim中添加以下內容:
blacklist nouveau options nouveau modeset=0 |
更新,重啓:
sudo update-initramfs -u sudo reboot |
驗證是否已經禁止nouveau nvidia驅動:
lsmod | grep nouveau |
4)進入文本模式安裝驅動:
使用Ctrl+Alt+F4命令切換到文本模式,切換至你下載驅動(.run)的路徑下安裝:
sudo ./NVIDIA.run |
確認是否安裝成功(根據左上角的NVIDIA-SMI即你下載的版本號對應):
nvidia-smi nvidia-settings |
2.3 Ubuntu下安裝CUDA驅動
1)GCC降級:
因爲CUDA 9.0僅支持GCC 6.0及如下版本,而Ubuntu 18.04預裝GCC版本爲7.3,下載GCC-4.8,g++-4.8版本,並切換至該版本。
sudo apt-get install gcc-4.8 sudo apt-get install g++-4.8
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 100 sudo update-alternatives --config gcc |
2)安裝CUDA及其補丁:
sudo sh cuda_10.1.85_387.26_linux.run sudo sh cuda_10.1.85.1_linux.run sudo sh cuda_10.1.85.2_linux.run sudo sh cuda_10.1.85.3_linux.run |
2.3 Ubuntu下安裝cuDNN驅動
直接在相應目錄下執行命令進行安裝(下例版本不對):
sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
|
驗證cuDNN是否安裝成功:
cp -r /usr/src/cudnn_sample_v7/ $HOME cd $HOME/cudnn_sample_v7/mnistCUDNN make clean && make ./mnistCUDNN
|
2.4 Ubuntu下安裝TensorFlow-gpu==1.14.0
建議先安裝virtualenv,而後在虛擬環境下安裝TensorFlow,固然也能夠直接安裝TensorFlow。
安裝過程:https://www.cnblogs.com/SsoZhNO-1/p/11177633.html
當你進入虛擬環境後,安裝TensorFlow-GPU:
# Source ./activate Pip install tensorflow-gpu==1.14.0 |
2.5Ubuntu18.04安裝cuda8
在安裝cuda8和cuda9的時候,都沒有對應Ubuntu18的run文化,對於cuda9咱們能夠直接安裝ubuntu17,在安裝多個版本的時候。選擇過程以下圖:
其中在安裝cuda8的時候可能會出現以下錯誤:
解決方法:
log中給出了幾個目錄的位置,而且說明在這些目錄下都沒法找到InstallUtils.pm(can't locate InstalUtils.pm)
選取多個目錄中的一個,先記錄下來。
我選取的一個是目錄是 /usr/lib/x86_64-linux-gnu/perl5/5.26
打開終端,進入放着cuda 8.0 runfile安裝文件的路徑下面,即 cuda_8.0.61_375.26_linux.run 的路徑下。
具體操做方法以下:
1)解壓runfile文件
$ sh cuda_8.0.61_375.26_linux.run --noexec --target 001 #將runfile文件解壓而且放到001,文件夾中(001可自動建立
|
2)進入解壓後的文件夾,將InstallUtils.pm拷貝到以前咱們從log中記錄的文件目錄下,注意可能使用sudo以得到管理員權限
$ sudo cp InstalUtil.pm ***** #拷貝複製,*****指目標目錄,這裏我使用的是/usr/lib/x86_64-linux-gnu/perl5/5.26 |
3)輸入export $PERL5LIB
$ export $PERL5LIB |
再次嘗試運行cuda 8.0 的runfile文件,安裝成功!
參考:https://blog.csdn.net/u014529295/article/details/78820414
!
在安裝多版本cuDNN的時候(原來是7.4,如今換成了6):
至此,版本從1.14.0退回到了1.4.0
切換CUDA版本
sudo rm -rf /usr/local/cuda #刪除以前建立的軟連接 sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda #建立新 cuda 的軟連接
|
Roi_pooling動態庫編譯
1、環境變量
OS:Ubuntu18.04
編程語言:python3.6
深度學習框架:TensorFlow-GPU==1.14.0
2、安裝步驟
Bash Code/DeepLogo_v2.1_gpu/Lib/make.sh |
(該腳本代碼和原代碼不一樣,已經被我改爲適應該版本代碼)