主機(jī)回來(lái)以及，簡(jiǎn)單的環(huán)境配置(RTX3070+CUDA11.1+CUDNN+TensorRT)

頭號(hào)碼甲 2022-08-04 發(fā)布于北京

展開(kāi)全文

緊接著前幾天的事：

特殊的日子，想起了當(dāng)年的雙(1080TI)顯卡裝機(jī)實(shí)錄
和
煉丹爐買不起了:聊一聊這段日子的顯卡行情

之后，決定買一臺(tái)整機(jī)玩玩。

而現(xiàn)在，主機(jī)終于回！來(lái)！了！主機(jī)回來(lái)干什么，當(dāng)然是——配置環(huán)境。

老潘之前也有一些配置環(huán)境的文章，可以參考：

-_-|| 沒(méi)想到那會(huì)寫了這么多配置環(huán)境的文章，可能那會(huì)遇到的問(wèn)題比較多吧...配環(huán)境什么的，其實(shí)只要嚴(yán)格按照步驟來(lái)一步一步配，基本上不會(huì)出現(xiàn)什么問(wèn)題，出現(xiàn)問(wèn)題一般都是我們某一步?jīng)]有走對(duì)，而退回去再走比較麻煩而已。

而現(xiàn)在對(duì)于配環(huán)境來(lái)說(shuō)可以是輕車熟路了(前幾年不知道踩了多少坑，到現(xiàn)在都印象深刻)，嚴(yán)格按照步驟來(lái)，安裝Ubuntu和配置深度學(xué)習(xí)環(huán)境，1個(gè)小時(shí)多就搞定了。

你懂得

接下來(lái)略微詳細(xì)地講一下過(guò)程。

Windows下安裝Ubuntu-18.04
配置深度學(xué)習(xí)環(huán)境(Cuda+Cudnn+Pytorch+TensorRT)

主機(jī)長(zhǎng)這樣

先放一組圖吧。

主機(jī)正面

躺下的主機(jī)

微星魔龍RTX3070正面

背部引線以及電源

插上顯卡的主機(jī)

晚上開(kāi)機(jī)長(zhǎng)這樣

晚上開(kāi)機(jī)還挺炫的，但其實(shí)對(duì)于老潘來(lái)說(shuō)炫不炫不重要，性能好就行...

主機(jī)配環(huán)境

主機(jī)回來(lái)店家只給我裝了Windows系統(tǒng)，對(duì)于老潘來(lái)說(shuō)，打游戲是次要的(哈哈哈哈...)，忍住了先下載一個(gè)鬼泣5玩玩的想法。先從Ubuntu官方下載了官方的鏡像包。

官方最新的Ubuntu是20.10，但對(duì)于搞事的人來(lái)說(shuō)，還是18.04版本好一些(之前實(shí)驗(yàn)室用的是16.04)。

還是下載穩(wěn)定版的吧

開(kāi)始配置雙系統(tǒng)，雙系統(tǒng)是必須的，Ubuntu可以搞深度學(xué)習(xí)，也可以當(dāng)服務(wù)器使，Windows則可以應(yīng)付一些游戲和其他應(yīng)急應(yīng)用。

關(guān)于如何將下載好的Ubuntu系統(tǒng)鏡像制作成U盤、如何安裝，老潘這里不贅述了?？梢钥匆韵逻@篇文章，介紹的很詳細(xì)，我就是按照這個(gè)嚴(yán)格來(lái)執(zhí)行的：

Windows下安裝Ubuntu詳細(xì)教程

當(dāng)然如果有問(wèn)題歡迎交流，直接留言即可~

安裝過(guò)程

雖然不贅述了，但還是簡(jiǎn)單過(guò)一下：

進(jìn)入BIOS，選擇Ubuntu啟動(dòng)U盤，然后開(kāi)始安裝Ubuntu：

其他一路點(diǎn)點(diǎn)點(diǎn)，然后簡(jiǎn)單分個(gè)區(qū)。

ubuntu分區(qū)

然后開(kāi)始安裝...

安裝ubuntu

等待半小時(shí)，裝好了！

設(shè)置SSH

為什么要設(shè)置ssh，當(dāng)然是想讓這個(gè)Ubuntu充當(dāng)一個(gè)服務(wù)器的作用，開(kāi)啟之后可以使用ssh登錄去操作。

例如我，可以先把服務(wù)器開(kāi)了，然后使用其他電腦，例如MAC，使用局域網(wǎng)通過(guò)SSH連接這個(gè)服務(wù)器即可。

怎么開(kāi)啟ssh呢？新的Ubuntu系統(tǒng)還沒(méi)有安裝SSH。

執(zhí)行以下命令：

sudo apt install openssh-server
sudo systemctl start ssh.service

然后可以通過(guò)netstat -lnp | grep 22查看下開(kāi)啟沒(méi)。

如果想要每次啟動(dòng)自動(dòng)開(kāi)啟SSH，可以這樣：

sudo systemctl enable ssh

這樣就差不多啦。

安裝NVIDIA顯卡驅(qū)動(dòng)

默認(rèn)Ubuntu安裝的是llvmpipe這個(gè)顯卡驅(qū)動(dòng)，這個(gè)是linux下的公用顯卡驅(qū)動(dòng)，現(xiàn)在需要換成NVIDIA的。

首先禁用nouveau。

執(zhí)行sudo gedit /etc/modprobe.d/blacklist.conf

加上以下這兩句：

blacklist nouveau
options nouveau modest=0

禁用noueavu

保存后，然后執(zhí)行：

sudo updata-initramfs -u
sudo reboot

重啟后，Ctrl+Alt+F1切換到tty界面，關(guān)閉lightdm(如果沒(méi)有則不用管)：

sudo service lightdm stop

然后更新一下apt源以及看一下系統(tǒng)推薦的NVIDIA驅(qū)動(dòng)版本：

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

根據(jù)推薦的驅(qū)動(dòng)版本，安裝NVIDIA驅(qū)動(dòng)：

sudo apt-get install nvidia-driver-460

如果嫌慢，可以添加阿里或者清華源：

sudo cp /etc/apt/sources.list /etc/apt/sources.list.bcakup
sudo gedit /etc/apt/sources.list

備份之后打開(kāi)文件，添加以下源即可：

# 阿里云源
deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
##測(cè)試版源
deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
# 源碼
deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
##測(cè)試版源
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse


# 清華大學(xué)源
deb http://mirrors.tuna./ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.tuna./ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.tuna./ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.tuna./ubuntu/ bionic-backports main restricted universe multiverse
##測(cè)試版源
deb http://mirrors.tuna./ubuntu/ bionic-proposed main restricted universe multiverse
# 源碼
deb-src http://mirrors.tuna./ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.tuna./ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.tuna./ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.tuna./ubuntu/ bionic-backports main restricted universe multiverse
##測(cè)試版源
deb-src http://mirrors.tuna./ubuntu/ bionic-proposed main restricted universe multiverse

別忘了添加之后，進(jìn)行更新二連：

sudo apt-get update
sudo apt-get upgrade

這樣就可以順利的安裝NVIDIA驅(qū)動(dòng)了。

然后下載好以下三個(gè)深度學(xué)習(xí)伴侶：

cuda+TensorRT環(huán)境包

開(kāi)始安裝吧！

安裝Cuda、Cudnn

Cuda當(dāng)然是必須要裝的。

找到下載好的11.1cuda環(huán)境包(現(xiàn)在出來(lái)11.2了)，然后執(zhí)行：

sudo sh cuda_11.1.0_455.23.05_linux.run

會(huì)出來(lái)一堆blabla的選項(xiàng)：

1、確保你環(huán)境里頭是否有舊的cuda，有的話建議刪除
2、同意條款..
3、確認(rèn)是否要安裝驅(qū)動(dòng)、安裝demo...以及安裝位置確定
4、開(kāi)始安裝ing

對(duì)于老潘來(lái)說(shuō)，上一步中已經(jīng)安裝了cuda驅(qū)動(dòng)，不需要再裝舊版的(新版驅(qū)動(dòng)兼容舊版驅(qū)動(dòng)匹配的cuda)，所以這里就把驅(qū)動(dòng)選項(xiàng)去掉，其他的安裝按照我的需求按部就班來(lái)就好。

cuda安裝選項(xiàng)

注意
如果你沒(méi)有root權(quán)限，無(wú)法使用sudo，想要安裝cuda也是可以的。只要自定義好安裝位置即可，運(yùn)行如下命令即可將cuda安裝到當(dāng)前home下的software文件夾內(nèi)：

./cuda_11.1.0_455.23.05_linux.run --silent --toolkit --toolkitpath=$HOME/software/cuda --defaultroot=$HOME/software/cuda

不論咋樣安裝好之后會(huì)顯示：

安裝好cuda的顯示

按照上面的要求配置環(huán)境變量即可:

(base) oldpan@oldpan-fun:~/software$ vim ~/.bashrc  

將
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
添加到打開(kāi)的文件中

(base) oldpan@oldpan-fun:~/software$ source ~/.bashrc
(base) oldpan@oldpan-fun:~/software$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Tue_Sep_15_19:10:02_PDT_2020
Cuda compilation tools, release 11.1, V11.1.74
Build cuda_11.1.TC455_06.29069683_0

Cudnn

cudnn的安裝就比較簡(jiǎn)單了，解決壓縮包然后復(fù)制粘貼就行：

tar -xzvf cudnn-11.2-linux-x64-v8.1.1.33.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h

安裝Anaconda

安裝Anaconda也很簡(jiǎn)單，從這里下載好(不想下的看老潘整理文末的軟件包)：

然后sh Anaconda3-2020.11-Linux-x86_64.sh就行。

安裝好之后，配置conda以及pip清華源，參考以下文字即可：

安裝Pytorch

安裝Pytorch就比較簡(jiǎn)單了，如果不想自己編譯，直接從官方按照你的Cuda版本和Cudnn版本安裝即可：

https://download.pytorch.org/whl/torch_stable.html

安裝好Pytorch之后，測(cè)試一下cuda是否工作正常：

>>> import torch
>>> torch.cuda.is_available()
True
>>> torch.ones(1).cuda()
tensor([1.], device='cuda:0')
>>> torch.cudnn_is_acceptable(torch.ones(1).cuda())
True

OK~

TensorRT

TensorRT單獨(dú)發(fā)一篇文章來(lái)說(shuō)吧~

一些資源

配環(huán)境需要很多軟件包，例如：

Anaconda
Pytorch.whl
TensorRT
CUDA
CUDNN

有些可以在清華大學(xué)開(kāi)源軟件鏡像站下載，但是TensorRT和CUDA、CUDNN需要從官方下而且需要注冊(cè)而且很慢。

老潘整理了一些已經(jīng)下載好的軟件包。

公眾號(hào)內(nèi)回復(fù)015獲取，可以看看有沒(méi)有你需要的：

軟件包

想寫的還有很多，下篇再講(呼呼)。

如果有問(wèn)題歡迎留言，歡迎關(guān)注「oldpan博客」公眾號(hào)，老潘的全部家當(dāng)都在這里了。很愿意與你交朋友~

參考

https://www.cnblogs.com/masbay/p/10745170.html
https://blog.csdn.net/ZPeng_CSDN/article/details/96726436

撩我吧

如果你與我志同道合于此，老潘很愿意與你交流；
如果你喜歡老潘的內(nèi)容，歡迎關(guān)注和支持。
如果你喜歡我的文章，希望點(diǎn)贊

本站是提供個(gè)人知識(shí)管理的網(wǎng)絡(luò)存儲(chǔ)空間，所有內(nèi)容均由用戶發(fā)布，不代表本站觀點(diǎn)。請(qǐng)注意甄別內(nèi)容中的聯(lián)系方式、誘導(dǎo)購(gòu)買等信息，謹(jǐn)防詐騙。如發(fā)現(xiàn)有害或侵權(quán)內(nèi)容，請(qǐng)點(diǎn)擊一鍵舉報(bào)。

轉(zhuǎn)藏 分享

QQ空間 QQ好友新浪微博微信

獻(xiàn)花（0） +1

來(lái)自：頭號(hào)碼甲 > 《待分類》

舉報(bào)/認(rèn)領(lǐng)