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

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

展開全文

緊接著前幾天的事：

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

之后，決定買一臺整機玩玩。

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

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

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

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

你懂得

接下來略微詳細地講一下過程。

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

主機長這樣

先放一組圖吧。

主機正面

躺下的主機

微星魔龍RTX3070正面

背部引線以及電源

插上顯卡的主機

晚上開機長這樣

晚上開機還挺炫的，但其實對于老潘來說炫不炫不重要，性能好就行...

主機配環(huán)境

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

官方最新的Ubuntu是20.10，但對于搞事的人來說，還是18.04版本好一些(之前實驗室用的是16.04)。

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

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

關(guān)于如何將下載好的Ubuntu系統(tǒng)鏡像制作成U盤、如何安裝，老潘這里不贅述了。可以看以下這篇文章，介紹的很詳細，我就是按照這個嚴格來執(zhí)行的：

Windows下安裝Ubuntu詳細教程

當然如果有問題歡迎交流，直接留言即可~

安裝過程

雖然不贅述了，但還是簡單過一下：

進入BIOS，選擇Ubuntu啟動U盤，然后開始安裝Ubuntu：

其他一路點點點，然后簡單分個區(qū)。

ubuntu分區(qū)

然后開始安裝...

安裝ubuntu

等待半小時，裝好了！

設(shè)置SSH

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

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

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

執(zhí)行以下命令：

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

然后可以通過netstat -lnp | grep 22查看下開啟沒。

如果想要每次啟動自動開啟SSH，可以這樣：

sudo systemctl enable ssh

這樣就差不多啦。

安裝NVIDIA顯卡驅(qū)動

默認Ubuntu安裝的是llvmpipe這個顯卡驅(qū)動，這個是linux下的公用顯卡驅(qū)動，現(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(如果沒有則不用管)：

sudo service lightdm stop

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

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

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

sudo apt-get install nvidia-driver-460

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

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

備份之后打開文件，添加以下源即可：

# 阿里云源
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
##測試版源
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
##測試版源
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
##測試版源
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
##測試版源
deb-src http://mirrors.tuna./ubuntu/ bionic-proposed main restricted universe multiverse

別忘了添加之后，進行更新二連：

sudo apt-get update
sudo apt-get upgrade

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

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

cuda+TensorRT環(huán)境包

開始安裝吧！

安裝Cuda、Cudnn

Cuda當然是必須要裝的。

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

sudo sh cuda_11.1.0_455.23.05_linux.run

會出來一堆blabla的選項：

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

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

cuda安裝選項

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

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

不論咋樣安裝好之后會顯示：

安裝好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
添加到打開的文件中

(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的安裝就比較簡單了，解決壓縮包然后復(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也很簡單，從這里下載好(不想下的看老潘整理文末的軟件包)：

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

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

安裝Pytorch

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

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

安裝好Pytorch之后，測試一下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~