換了一臺電腦,搭建環境浪費了半天時間,現在記錄一下。
1.安裝Nvidia驅動
我之前在 https://www.geforce.cn/drivers 下載了適用於RTX 2080Ti的Nvidia 430,但是運行的時候報錯check sum xxxxx.
於是放棄了先下載再安裝,用了下面的方式:
Ctrl +Alt +F1進入命令行模式,輸入用戶名和密碼
sudo apt-get update
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall //會自動安裝適合的驅動
reboot//重啓
此時,可以看到驅動已經裝好了。
qogori@qogori-Nuvo-6108GC:~/c++$ nvidia-smi
Tue May 26 14:36:47 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.64 Driver Version: 430.64 CUDA Version: 10.1 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce RTX 208... Off | 00000000:01:00.0 On | N/A |
| 30% 40C P8 2W / 250W | 491MiB / 11011MiB | 2% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1015 G /usr/lib/xorg/Xorg 222MiB |
| 0 1740 G compiz 53MiB |
| 0 1972 G fcitx-qimpanel 15MiB |
| 0 2002 G ...AAAAAAAAAAAACAAAAAAAAAA= --shared-files 192MiB |
| 0 3416 G ...e/qogori/QtCreator/latest/bin/qtcreator 6MiB |
+-----------------------------------------------------------------------------+
2.安裝Cuda
下載地址:https://developer.nvidia.com/cuda-toolkit-archive
選擇下載cuda_10.0.130_410.48_linux.run,
sudo sh cuda_10.0.130_410.48_linux.run
一直按着ENTER進行安裝,直到需要輸入。
accept,安裝NVIDIA驅動選 no,別的都選yes即可
安裝完,修改環境變量:
sudo gedit ~/.bashrc
在最後面加下面的語句:
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
sudo reboot
接下來,測試是否安裝成功:
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
最後一行爲Result = PASS,即爲安裝成功。
注:因爲聽說darknet對cuda9.0更穩定,下載了cuda_9.0.176_384.81_linux.run,安裝完運行./deviceQuery,報錯如下:
cudaGetDeviceCount returned 35 -> CUDA driver version is insufficient for CUDA runtime version Result = FAIL
我查了半天,不知道是不是因爲 driver 版本過高了。有知道的大哥可以說說是爲啥。
3.安裝Cudnn
下載地址:https://developer.nvidia.com/rdp/cudnn-download
我選擇的是cudnn-10.0-linux-x64-v7.6.5.32.tgz。
tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.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 /usr/local/cuda/lib64/libcudnn*
查看版本:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
會出現如下內容:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
到這裏就安裝完畢了。
希望之後其它環境的安裝一切順利。