一. nvidia驱动安装

1. 查询显卡型号

nvidia-smi -L

出现显卡型号,我的是:GPU 0: GeForce RTX 2080 SUPER

2. 查询显卡驱动

网址:https://www.geforce.cn/drivers
输入显卡型号并搜索,列表中会出现很多驱动版本,选择一款驱动

3.安装显卡驱动

网上常见的做法是:从命令行系统安装,巨麻烦,推荐使用apt-get install

sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update

sudo apt-get install nvidia- table键补全,安装你要的驱动即可

sudo apt-get install nvidia-430

4.查看显卡驱动

nvidia-smi

二. cuda安装

cuda对于nvidia驱动是有要求的,具体版本需要请见:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
ubuntu 16.04安装nvdia驱动,cuda驱动以及cudnn
cuda驱动下载地址:https://developer.nvidia.com/cuda-toolkit-archive

注意cuda对gcc版本也是有要求的:

从CUDA 7.5版开始,完全支持gcc 4.8,在Ubuntu 14.04和Fedora 21上支持4.9。
从CUDA 8版本开始,Ubuntu 16.06和Fedora 23完全支持gcc 5.3。
从CUDA 9版本开始,Ubuntu 16.04,Ubuntu 17.04和Fedora 25完全支持gcc 6。

具体安装请参见, 图形界面安装即可:https://blog.csdn.net/briliantly/article/details/79560477

查看驱动是否安装成功:

nvcc -V

我这里输出的信息为:

nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2017 NVIDIA
Corporation Built on Fri_Nov__3_21:07:56_CDT_2017 Cuda compilation
tools, release 9.1, V9.1.85

三. cuDNN安装

cuDNN对cuda版本也有需求,具体参见:https://developer.nvidia.com/rdp/cudnn-archive
安装非常简单,复制粘贴即可:

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*

相关文章: