Ubuntu 16.04/18.04 安装GPU驱动+Cuda 10.1+cuDNN 7.6

GPU驱动,CUDA以及cuDNN一直在持续更新中,现在的安装也越来越简单!

1. 现在的CUDA安装包中自带的GPU驱动已经满足所有需求,不需要提前安装GPU驱动;所以直接从下边网址下载

https://developer.nvidia.com/cuda-downloads

Linux--x86_64--Ubuntu--16.04/18.04--runfile(local)

2. Ctrl+Alt+F1进入命令行模式,依次执行

sudo service lightdm stop
chomod +x 下载的run文件
sudo ./下载的run文件
sudo service lightdm start

安装过程中,所有部件都安装(默认全部安装),安装完之后,打开./bashrc文件:gedit ~/.bashrc,在末尾加上

export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH

然后保存文件,执行source命令: source ~/.bashrc

然后运行命令:nvidia-smi 检查GPU安装情况,nvcc -V 检查CUDA安装情况,会显示GPU和CUDA的版本。

3.安装cuDNN

打开网址https://developer.nvidia.com/rdp/cudnn-download

下载三个文件

cuDNN Runtime Library for Ubuntu16.04 (Deb)

cuDNN Developer Library for Ubuntu16.04 (Deb)

cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)

打开terminal,用dpkg -i 命令依次安装上述三个文件,一个例子如下:

sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-devel_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb

4.检查cuDNN安装情况

cp -r /usr/src/cudnn_samples_v7/ $HOME
cd  $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN

最终看到屏幕输出 Test passed! 表示一切安装顺利!

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值