cuda+cudnn+tensorrt

Ubuntu16.04 安装NVIDIA显卡驱动

驱动下载地址:https://www.geforce.cn/drivers

在安装显卡驱动之前,需要卸载之前安装的显卡驱动,如果你是刚安装的Ubuntu系统,就不用了。

首先我们先下载好我们需要的驱动文件,(我的并重命名为NVIDIA.run)并保存到根目录下,这样就省的再找了。

ctrl+alt+t

sudo apt-get remove --purge nvidia*

sudo gedit /etc/modprobe.d/blacklist.conf

在打开的文本编辑器的最下方加入下面两句

blacklist nouveau

options nouveau modeset=0

然后保存文件

ctrl+alt+t

sudo update-initramfs -u

重启

ctrl+alt+t

lsmod |grep nouveau

如果没有输出任何信息,就继续下一步

sudo service lightdm stop

此时会黑屏,不用担心,没事的,使用ctrl+alt+F1,输入用户名和密码(不要使用键盘的数字键盘输入数字,使用字母键盘上的数字键输入),验证通过后,

sudo chomd +x NVIDIA.run     #这里的NVIDIA.run是驱动的文件名,

sudo bash NVIDIA.run -no-opengl-files

sudo service lightdm start

在安装的过程中,需要手动的设置一下,再问到是否以后更新驱动时,选择默认的”否“

重启,

Ctrl+Alt+F7

使用nvidia-smi或nvidia-settings查看是否安装成功和驱动版本

Guide:https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

cuda:

1.转至下载文件路径,执行

sudo dpkg -i cuda-repo-.deb

后面的cuda-repo-.deb是下载的文件名。 注意各个部分的对应关系,后面用得到。

2.sudo apt-key add /var/cuda-repo-/7fa2af80.pub

3.sudo apt-get install cuda

cudnn:

a,安装runtime library

sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb

b,安装developer library

sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb

c,安装code samples and the cuDNN Library User Guide

sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb

TensorRT:

sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb

sudo apt-key add /var/nv-tensorrt-repo-${tag}/7fa2af80.pub

 

sudo apt-get update

sudo apt-get install tensorrt

 

你可能感兴趣的:(linux)