因?yàn)閁buntu18.04剛發(fā)布不久，在安裝CUDA和CUDNN中還沒有匹配版本，最高也是17.04的版本，但實(shí)際上18.04的版本具有很強(qiáng)的兼容性。為避免讀者踩坑，本文測(cè)試成功了Ubuntu18.04環(huán)境下配置深度學(xué)習(xí)環(huán)境（GPU:1080ti），包括：

CUDA9.0+CUDNN7.4+TensorFlow1.8+Pytorch0.4的安裝和測(cè)試，并提供相關(guān)軟件下載。

硬件配置

DELL R730服務(wù)器(機(jī)架式2U服務(wù)器)

顯卡?1080TI *2

內(nèi)存 64G

CPU 2620V4* 2

電源 DELL 1600w *2

硬盤 600G*2+2T*2

安裝

備注：安裝步驟提到的所有安裝包和驅(qū)動(dòng)，本文提供下載，在本文結(jié)尾處有下載地址。

1.安裝Ubuntu

使用U盤進(jìn)行Ubuntu操作系統(tǒng)的安裝：

參考：https://jingyan.baidu.com/article/a3761b2b66fe141577f9aa51.html

一開始安裝選擇"Install Ubuntu"回車后過一會(huì)兒屏幕如果顯示“輸入不支持”，這和Ubuntu對(duì)顯卡的支持有關(guān)，在安裝主界面的F6，選擇nomodeset，就可以進(jìn)入下一步安裝了

安裝過程略，安裝鏡像下載地址：

https://www.ubuntu.com/download/desktop

下載：ubuntu-18.04.1-desktop-amd64.iso

備注：如果下載不方便，本文末尾處提供百度云下載。

2.安裝ssh

備注：這一步需要到服務(wù)器桌面上的命令窗口輸入，這一步完成后，就可以用ssh工具遠(yuǎn)程連接服務(wù)器了，本文使用的是XShell。

sudo apt-get install openssh-server

3. 安裝1080TI顯卡驅(qū)動(dòng)

默認(rèn)安裝的顯卡驅(qū)動(dòng)不是英偉達(dá)的驅(qū)動(dòng)，所以先把舊得驅(qū)動(dòng)刪除掉。

sudo apt-get purgenvidia*

添加Graphic Drivers PPA

sudo add-apt-repository ppa:graphics-drivers/ppasudo apt-get update

查看合適的驅(qū)動(dòng)版本：

ubuntu-drivers devices

圖：可用的nvidia 驅(qū)動(dòng)列表

圖中可以看出推薦的是最新的415版本的驅(qū)動(dòng)，安裝該驅(qū)動(dòng)：

sudo apt-get install nvidia-driver-415

安裝完畢后重啟機(jī)器：

sudo reboot

重啟完畢運(yùn)行

nvidia-smi

看看生效的顯卡驅(qū)動(dòng)：

圖：生效的顯卡驅(qū)動(dòng)

4.安裝依賴庫(kù)

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-devlibgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

5.GCC降低版本

CUDA9.0要求GCC版本是5.x或者6.x，其他版本不可以，需要自己進(jìn)行配置，通過以下命令才對(duì)GCC版本進(jìn)行修改。

• 版本安裝

# 版本安裝： sudo apt-get install gcc-5 sudo apt-get install g++-5

• 通過命令替換掉之前的版本?
  

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 50 sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 50

6.安裝CUDA9.0

• 下載網(wǎng)址：

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfilelocal

• 版本選擇：

linux--x86-64，Ubuntu--17.04，runfile(local)，下載Base Installer和4個(gè)Patch

備注：如果下載不方便，本文末尾處提供百度云下載。

下載進(jìn)入目錄后，依次執(zhí)行命令：

注意：在安裝過程中會(huì)提示是否需要安裝顯卡驅(qū)動(dòng)，在這里要選擇n，其他的選擇y或者回車鍵進(jìn)行安裝:

sudo sh cuda_9.0.176_384.81_linux.runsudo sh cuda_9.0.176.1_linux.runsudo sh cuda_9.0.176.2_linux.runsudo sh cuda_9.0.176.3_linux.runsudo sh cuda_9.0.176.4_linux.run

• 在~/.bashrc 中設(shè)置環(huán)境變量：

sudo vi ~/.bashrc

在最后添加：

exportPATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} exportLD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

運(yùn)行

source ~/.bashrc

使其生效，最好重啟一下。

sudo reboot

重啟后，測(cè)試CUDA是否成功（此步驟可以省略）

cd ~/NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery make -j4 ? sudo ./deviceQuery

若會(huì)輸出相應(yīng)的顯卡性能信息，Result = PASS，表明CUDA安裝成功。

7.安裝CUDNN

https://developer.nvidia.com/rdp/cudnn-download

需要登錄，同意后才能下載。

備注：如果下載不方便，本文末尾處提供百度云下載。

點(diǎn)擊：

Download cuDNN v7.4.1(Nov 8, 2018), for CUDA 9.0 --cuDNNLibrary for Linux

下載時(shí)候保存為：cudnn-9.0-linux-x64-v7.4.1.5.tgz

下載后安裝非常簡(jiǎn)單，就是解壓然后拷貝到相應(yīng)的系統(tǒng)CUDA路徑下，注意最后一行拷貝時(shí) "-d"不能少，

tar -zxvf cudnn-9.0-linux-x64-v7.4.1.5.tgzsudo cp cuda/include/cudnn.h /usr/local/cuda/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r/usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

沒有報(bào)錯(cuò)就是全部安裝完成了。

8.安裝Tensorflow GPU 1.8

由于Anaconda可以提供完整的科學(xué)計(jì)算庫(kù)，所以直接使用Anaconda來進(jìn)行相關(guān)的安裝。

1）安裝Anaconda

下載地址：https://www.anaconda.com/download/

備注：如果下載不方便，本文末尾處提供百度云下載。

這里我們下載Python 3.7 64bit 的Anaconda3-5.3.0-Linux-x86_64.sh，直接安裝即可。

bash Anaconda3-5.3.0-Linux-x86_64.sh

2）更改pip和conda為國(guó)內(nèi)的源

由于國(guó)內(nèi)訪問pip和conda比較慢，建議更改為國(guó)內(nèi)的源：

a.更改pip的源為阿里云：

mkdir ~/.pipcat > ~/.pip/pip.conf << EOF [global] trusted-host=mirrors.aliyun.com index-url=https://mirrors.aliyun.com/pypi/simple/ EOF

b.更改conda的源為清華大學(xué)：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yes

3）在Anaconda中安裝Python3.6的虛擬環(huán)境

由于Anaconda的python版本是3.7的，TensorFlow尚不支持此版本，于是我們創(chuàng)建一個(gè)Python的虛擬環(huán)境

conda create --name tf python=3.6 #創(chuàng)建tf環(huán)境

虛擬環(huán)境主要命令：

source activate tf? ? ? ? ? ? ?#激活tf環(huán)境source deactivate tf? ? ? ? ? #退出tf環(huán)境conda remove --name tf --all? #刪除tf環(huán)境（全部刪除）

4）在Anaconda中安裝TensorFlow GPU 1.8

pip install --ignore-installed --upgrade tensorflow-gpu

將會(huì)自動(dòng)安裝如下組件：

numpy?、wheel?、tensorflow-tensorboard?、six、protobuf?、html5lib?、markdown、werkzeug?、bleach、setuptools

5）使用下列代碼測(cè)試安裝正確性

命令行輸入：

source activate tfpython

Python命令下輸入以下代碼：

import tensorflowas tf hello= tf.constant('Hello, TensorFlow!') sess= tf.Session() print(sess.run(hello))

沒有報(bào)錯(cuò)就是配置好了。

9.安裝Keras

直接在這個(gè)虛擬環(huán)境中安裝：

pip install keras

10.安裝Pytorch

直接在這個(gè)虛擬環(huán)境中安裝：

conda install pytorch torchvision -c pytorch

測(cè)試Pytorch是否安裝成功：

命令行輸入：

source?activate?tfpython

python命令下輸入以下代碼：

import torch print(torch.cuda.is_available())

?

返回True說明安裝成功了。

總結(jié)

因?yàn)閁buntu18.04剛發(fā)布不久，用戶相對(duì)比較少，為避免讀者踩坑，本文測(cè)試成功了Ubuntu18.04環(huán)境下配置深度學(xué)習(xí)環(huán)境（GPU:1080ti），包括：

CUDA9.0+CUDNN7.4+TensorFolw1.8+Pytorch0.4的安裝和測(cè)試，并提供相關(guān)軟件的百度云下載。

作者：黃海廣

參考

https://blog.csdn.net/weixin_41863685/article/details/80303963

本文所有需要的軟件下載請(qǐng)回復(fù)“1804”查看

也可以直接用百度云下載：

鏈接：https://pan.baidu.com/s/11AxudhGg3KicM8NzBjvFiw?

提取碼：w214?

請(qǐng)關(guān)注和分享↓↓↓?

機(jī)器學(xué)習(xí)初學(xué)者

QQ群：654173748

往期精彩回顧

• 機(jī)器學(xué)習(xí)簡(jiǎn)易入門-附推薦學(xué)習(xí)資料

• 機(jī)器學(xué)習(xí)初學(xué)者公眾號(hào)下載資源匯總（一）
  

• 黃海廣博士的github鏡像下載（機(jī)器學(xué)習(xí)及深度學(xué)習(xí)資源）

• 吳恩達(dá)老師的機(jī)器學(xué)習(xí)和深度學(xué)習(xí)課程筆記打印版
  

• 機(jī)器學(xué)習(xí)小抄-（像背托福單詞一樣理解機(jī)器學(xué)習(xí)）

• 首發(fā)：深度學(xué)習(xí)入門寶典-《python深度學(xué)習(xí)》原文代碼中文注釋版及電子書

• 科研工作者的神器-zotero論文管理工具

• 機(jī)器學(xué)習(xí)的數(shù)學(xué)基礎(chǔ)

總結(jié)

以上是生活随笔為你收集整理的Ubuntu 18.04深度学习环境配置（CUDA9.0+CUDDN7.4+TensorFolw1.8）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。