Ubuntu16.04+RTX2070s 安裝顯卡驅(qū)動(dòng)+CUDA+CUDNN+Pytorch

安裝Ubuntu之后由于沒(méi)有顯卡驅(qū)動(dòng),默認(rèn)使用了Nouveau,顯示器分辨率極低

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

一般可以使用ppa源安裝或者自己從官網(wǎng)下,我在使用ppa添加源的時(shí)候一直沒(méi)反應(yīng),可能是網(wǎng)絡(luò)原因,所以這里選擇了第二種:

從NVIDIA官網(wǎng):https://www.geforce.cn/drivers選擇合適的版本下載,放在英文目錄下(有可能在文本模式下沒(méi)法顯示中文)

如果之前有安裝驅(qū)動(dòng),需要先卸載舊版本驅(qū)動(dòng),沒(méi)有就不用:

$sudo apt-get remove nvidia-*

$sudo apt-get autoremove? (加上才能卸載干凈原驅(qū)動(dòng))

輸入CTRL+ALT+F1進(jìn)入文本模式

關(guān)閉顯示服務(wù):

$sudo service lightdm stop

權(quán)限:

$chmod a+x NVIDIA-Linux-x86_64-418.43.run

這里注意用自己的安裝包名稱

$sudo ./NVIDIA-Linux-x86_64-418.43.run -no-x-check -no-nouveau-check -no-opengl-files

以下三個(gè)最好加上,否則還是可能循環(huán)登錄

? ? -no-x-check安裝驅(qū)動(dòng)時(shí)關(guān)閉x服務(wù);

? ? -no-nouveau-check 安裝驅(qū)動(dòng)時(shí)禁用Nouveau

? ? -no-opengl-files 安裝時(shí)只裝驅(qū)動(dòng)文件,不安裝Opengl

關(guān)于禁用Nouveau,還有一種方法是在/etc/modprobe.d/blacklist.conf的最后一行加上

blacklist nouveau

然后使其生效:

$sudo update-initramfs -u

使用這個(gè)命令看是否有輸出,如果沒(méi)有證明禁用成功

$lsmod | grep nouveau

我兩種都做了,目前沒(méi)有出現(xiàn)什么問(wèn)題

安裝過(guò)程中:

在NVIDIA驅(qū)動(dòng)安裝過(guò)程中,依次的選項(xiàng)為:

選:accept(注:白色為選中,不是紅色)

The distribution-provided pre-install script failed … …

選:Continue installation

Would you like to run the nvidia-xconfig utility to automatically update your X Configuration file so set the NVIDIA X driver will be used when you restart X?

選:NO

Install 32-Bit compatibility libraries?

選:NO

啟動(dòng)顯示服務(wù)

$sudo service lightdm restart(將自動(dòng)跳轉(zhuǎn)到桌面)

查看Nvidia驅(qū)動(dòng)是否安裝成功

$nvidia-smi

顯示信息的話表示安裝成功

2、安裝CUDA和CUDNN

CUDA:

去英偉達(dá)官網(wǎng)下載cuda安裝包https://developer.nvidia.com/cuda-toolkit-archive

跟據(jù)你的平臺(tái)選擇好后會(huì)出現(xiàn)安裝命令,依照其執(zhí)行即可,比如,

選擇合適的版本,我選擇了.run形式的安裝包,下載好后執(zhí)行

$sudo sh cuda_10.1.243_418.87.00_linux.run

在安裝過(guò)程中,會(huì)讓你選擇是否安裝驅(qū)動(dòng),這里有于我們已經(jīng)裝過(guò)了,所以選擇不裝驅(qū)動(dòng)

然后設(shè)置環(huán)境:在~/.bashrc文件最后添加下面幾行

LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64

PATH=$PATH:/usr/local/cuda-10.1/bin

CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

然后輸入

$nvcc --version

如果出現(xiàn)如下版本信息,證明安裝成功:

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2019 NVIDIA Corporation

Built on Sun_Jul_28_19:07:16_PDT_2019

Cuda compilation tools, release 10.1, V10.1.243

也可以這樣進(jìn)行測(cè)試:

$cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery

$sudo make

$sudo ./deviceQuery

如果最后輸出PASS,則安裝成功

CUDNN:

依然是官網(wǎng)下載安裝包:https://developer.nvidia.com/rdp/cudnn-archive

這里選擇了對(duì)應(yīng)CUDA版本的對(duì)應(yīng)linux版,文件格式是.solitairetheme8,需要先轉(zhuǎn)成tgz格式再解壓,直接重命名即可。

解壓下載的文件cuda文件夾,打開(kāi)終端,執(zhí)行如下命令:

$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

$sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

查看cudnn版本信息:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

出現(xiàn)版本信息則安裝成功

#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"

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容