码迷,mamicode.com
首页 > 系统相关 > 详细

Ubuntu18.04上安装N卡驱动、CUDA、CUDNN三连

时间:2019-05-09 23:48:59      阅读:357      评论:0      收藏:0      [点我收藏+]

标签:upgrade   安装ubuntu   dev   变量   目录   https   local   ade   注册   

环境:Ubuntu18.04

显卡驱动真的挺方便的,CUDA和CUDNN还是踩了一些坑2333

 

1、安装显卡驱动
安装ubuntu更新或sudo apt-get update & sudo apt-get upgrade

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall

安装显卡驱动

sudo reboot
nvidia-smi

若输出显卡信息则安装成功

非常方便了

 

2、安装CUDA
在官网https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal下载安装包,默认下载到Downloads目录下

cd ~/Downloads/

运行下载的.run文件,如:cuda_10.1.105_418.39_linux.run

sudo sh cuda_*_linux.run

在过程中,选择需要安装的内容时,请将驱动driver去掉,不要安装显卡驱动!
配置环境变量

export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
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

查看CUDA版本

cat /usr/local/cuda/version.txt

 

3、安装CUDNN
在官网https://developer.nvidia.com/cudnn注册账号并下载,选择下载cuDNN Library for Linux,不过说实话现在官网给的版本好多好多看花了眼23333,默认下载到Downloads目录下,解压并复制

cd ~/Downloads/
tar -xzvf cudnn-*
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*

查看CUDNN版本

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

 

Ubuntu18.04上安装N卡驱动、CUDA、CUDNN三连

标签:upgrade   安装ubuntu   dev   变量   目录   https   local   ade   注册   

原文地址:https://www.cnblogs.com/cenariusxz/p/10841628.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!