深度學(xué)習(xí)

Author：louwill

Machine?Learning Lab

? ?

???? 搞深度學(xué)習(xí)環(huán)境永遠(yuǎn)是第一步。筆者之前也寫過(guò)配置的兩篇文章，但時(shí)間久遠(yuǎn)，目前來(lái)看版本已經(jīng)過(guò)舊了。之前兩篇參考：

深度學(xué)習(xí)100問(wèn)-1：深度學(xué)習(xí)環(huán)境配置有哪些坑？

深度學(xué)習(xí)筆記15：ubuntu16.04 下深度學(xué)習(xí)開發(fā)環(huán)境搭建與配置

???? 顯卡與服務(wù)器等硬件設(shè)備裝機(jī)完成后就需要配置本機(jī)環(huán)境。本篇配置主要包括Nvidia驅(qū)動(dòng)安裝、CUDA安裝和深度學(xué)習(xí)與Python環(huán)境配置三個(gè)方面。本機(jī)系統(tǒng)配置為Ubuntu18.04+CUDA10.2，顯卡信息為GeForce GTX 1080Ti。

1.安裝Nvidia顯卡驅(qū)動(dòng)

???? 查看本機(jī)顯卡能夠配置的驅(qū)動(dòng)信息：

ubuntu-drivers devices

???? 可以看到本機(jī)推薦的驅(qū)動(dòng)版本為440，然后根據(jù)Nvidia官網(wǎng)上CUDA10.2 的配置要求看一下是否匹配：

???? 官方給出的CUDA10.2最低驅(qū)動(dòng)版本為440.33，所以要安裝CUDA10.2的話，驅(qū)動(dòng)器版本只有440可選。依次執(zhí)行下述命令安裝440驅(qū)動(dòng)：

sudo?apt-get?purge?nvidia*? sudo?add-apt-repository?ppa:graphics-drivers/ppa? sudo?apt-get?update? sudo apt-get install nvidia-driver-440

???? 然后重啟機(jī)器使更新生效：

sudo shutdown -r now

2.安裝CUDA10.2

???? 第二步就是安裝CUDA。這里我們安裝CUDA最新的10.2版本。進(jìn)入Nvidia cuda下載地址：

https://developer.nvidia.com/cuda-downloads

???? 依次選擇對(duì)應(yīng)系統(tǒng)和版本等信息：

???? 我們這里選擇的Ubuntu 18.04下的deb安裝類型。然后依次執(zhí)行下述7條命令：

wget?https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin sudo?mv?cuda-ubuntu1804.pin?/etc/apt/preferences.d/cuda-repository-pin-600 wget?http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb sudo?dpkg?-i?cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb sudo?apt-key?add?/var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub sudo?apt-get?update sudo apt-get -y install cuda

???? 如果順利的話CUDA10.2可以安裝完成。但如果本機(jī)之前的CUDA版本存在卸載不干凈等問(wèn)題的話，安裝有可能會(huì)報(bào)dkpg的錯(cuò)：

???? 所以這里補(bǔ)充一個(gè)CUDA完全卸載的方法：

sudo?apt-get?--purge?remove?"*cublas*"?"cuda*" sudo?apt-get?--purge?remove?"*nvidia*" sudo?apt-get?purge?nvidia* sudo?apt-get?autoremove sudo?apt-get?autoclean sudo rm -rf /usr/local/cuda*

3.安裝Python3.7 并配置深度學(xué)習(xí)環(huán)境

???? 下載anaconda：

wget https://repo.anaconda.com/archive/Anaconda3-2020.02-Linux-x86_64.sh

???? 安裝：

bash Anaconda3-2020.02-Linux-x86_64.sh

???? 創(chuàng)建虛擬環(huán)境：

conda install -n 虛擬環(huán)境名稱

???? 啟動(dòng)虛擬環(huán)境：

conda activate 虛擬環(huán)境名稱

???? 安裝深度學(xué)習(xí)框架，以PyTorch1.5為例：

pip install torch torchvision

???? 安裝完成后驗(yàn)證一下當(dāng)前環(huán)境下的PyTorch是否可以使用GPU，顯示為True則表明配置成功。

import?torch torch.cuda.is_available()

往期精彩回顧適合初學(xué)者入門人工智能的路線及資料下載機(jī)器學(xué)習(xí)及深度學(xué)習(xí)筆記等資料打印機(jī)器學(xué)習(xí)在線手冊(cè)深度學(xué)習(xí)筆記專輯AI基礎(chǔ)下載（pdf更新到25集）機(jī)器學(xué)習(xí)的數(shù)學(xué)基礎(chǔ)專輯本站qq群1003271085，加入微信群請(qǐng)回復(fù)“加群”獲取一折本站知識(shí)星球優(yōu)惠券，復(fù)制鏈接直接打開：https://t.zsxq.com/yFQV7am喜歡文章，點(diǎn)個(gè)在看

總結(jié)

以上是生活随笔為你收集整理的Ubuntu18.04+CUDA10.2 深度学习开发环境配置指南的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

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