cuda 安装_ubuntu16.04+Geforce GTX1060安装CUDA和cuDNN

DELL G7系列,ubuntu16.04,本机显卡型号: GeForce GTX 1060。

主要步骤如下:

一,安装NVIDIA驱动。
1,本笔记本是F12进入BIOS,将secure boot设置为disable。

2,通过下面命令确保电脑上之前的nvidia驱动已卸载干净(可选)。

sudo apt-get remove --purge nvidia*

3,通过下面命令卸载之前下载的驱动安装包(runfile格式)(可选)。

sudo sh /home/xxx/驱动名.run --uninstall  

4,屏蔽开源驱动nouveau,这里选择blacklist,首先创建新文件。

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

内容如下:

blacklist nouveau
options nouveau modeset=0

再通过如下命令更新:

sudo update-initramfs -u

通过如下命令,nouveau驱动是否被禁用(无输出则被禁用)。

lsmod | grep nouveau

5,打开终端加入,加入官方ppa源。

sudo add-apt-repository ppa:graphics-drivers/ppa

6,刷新软件库。

sudo apt-get update

7,查看推荐的显卡驱动,选择驱动后边有 推荐(recommended)字符的。

ubuntu-drivers devices

v2-c49e59aea34138d6d1ce604b772d5c08_b.jpg
本机是推荐nvidia-430,建议不在NVIDIA官网下最新驱动。

8,通过下面命令安装推荐的显卡驱动,并在完成后重启。

sudo apt-get install nvidia-430 nvidia-settings nvidia-prime
reboot

9,检查(确保两个命令都有正常输出)。

下面命令显示显卡的一些内容信息。

nvidia-smi

设置显卡。

nvidia-settings

二,安装cuda和cudnn

1,cuda官网下载,https://developer.nvidia.com/cuda-10.0-download-archive,选择runfile版本。

2,将run文件拷贝到HOME下面,在shell下,HOME目录里执行如下命令。将cuda_10.0.130_410.48_linux.run重命名为cuda.run。

sudo sh cuda.run

这里会显示用户认证信息,按回车键加载到100%。先accept,INSTALL NVIDIA ACCELERATED GRAPHICS DRIVER............选择no。

后面就一路yes和回车即可。

3,设置环境变量。

sudo gedit /etc/profile

并在文件末尾添加如下两行。

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

并执行如下命令,保证了环境变量的永久配置。

source /etc/profile
reboot

4,验证驱动版本和CUDA版本。

cat /proc/driver/nvidia/version

v2-0015114d4cbf8165d3ac89314f067321_b.png
nvcc -V

v2-b2eba521884f1c24301fef06103c840d_b.jpg

5,安装cudnn

cuDNN Archive​developer.nvidia.com

需要先注册NVIDIA帐号,并通过如下命令解压(在下载好的目录里)。

tar -xvf cudnn-10.0-linux-x64-v7.3.1.20.tgz

解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令。

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

参考链接:

Ubuntu16.04上CUDA_10.0及cuDNN的安装_lihe的博客-CSDN博客​blog.csdn.net
v2-2a5027b5bff83f50a189c6146b4f7548_ipico.jpg
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值