Ubuntu14.04安装CUDA8.0与Cudnn5.1

最新推荐文章于 2020-01-02 15:35:08 发布

转载最新推荐文章于 2020-01-02 15:35:08 发布 · 131 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/aiuestcer/p/6755700.html

文章标签：

#python #人工智能 #操作系统

本文详细介绍了在Linux环境下安装CUDA的过程，包括关闭nouveau驱动、安装CUDA run文件、配置环境变量等步骤，并提供了验证安装成功的命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

版权声明：本文为博主原创文章，未经博主允许不得转载。

一、安装CUDA

1.参考资料：官方文档 NVIDIA CUDA INSTALLATION GUIDE FOR LINUX

2.安装方式

官方文档中有deb与run两种安装方式，我选择run安装方式

选择理由：The Runfile installation installs the NVIDIA Driver, CUDA Toolkit, and CUDA Samples
via an interactive text-based interface.一站式安装 NVIDIA Driver, CUDA Toolkit, and CUDA Samples，方便快捷！

3.安装前准备

跟着官方文档的Pre-installation Actions走就对了，一般不会出现什么问题的！

4.关闭nouveau

4.1

$sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
           blacklist nouveau
           options nouveau modeset=0

4.2 Regenerate the kernel initramfs:

$sudo update-initramfs -u（这一步很关键，如果跳过直接重启，则nouveau关闭失败，看了官方文档之后才知道了，看官方文档的重要性）

4.3 重启

$sudo reboot

4.4 检验是否关闭nouveau

$lsmod | grep nouveau

若输入上述命令之后终端没有任何显示，则成功关闭nouveau

5.安装Runfile

5.1 Ctrl +F1+Alt

5.2 输入自己的用户名与密码

5.3

$sudo service lightdm stop

5.4 安装我所下载的版本，不同版本注意修改

$sudo sh cuda_8.0.61_375.26_ linux.run

5.5 环境变量设置

*官方文档此步为重启，我配置完环境变量之后再重启

$sudo gedit /etc/proflie
           export PATH=/usr/local/cuda-8.0/bin:$PATH
           export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH（输入内容）
$source /etc/profile

$sudo gedit /etc/ld.so.conf.d/cuda.conf 
           /usr/local/cuda/lib64（输入内容）
$sudo ldconfig

5.6 重启

$sudo service lightdm start
$sudo reboot

6.检验安装CUDA是否成功

$nvidia-smi

如果出现NVIDIA相关信息，则表明安装成功！

二、安装Cudnn

1.参考资料：http://blog.youkuaiyun.com/u013645510/article/details/53448806

2.解压

$tar xzvf cudnn-8.0-linux-x64-v5.1.tgz

我所下载的cudnn版本，我下载在主文件夹里边，直接解压，如果下载到其他文件夹则cd到那个文件夹目录下在解压.

3.安装

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
$ cd /usr/local/cuda/lib64
$ sudo rm -rf libcudnn.so libcudnn.so.5
$ sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5(不同版本的cudnn，libcudnn.so.5.1.x不同)
$ sudo ln -s libcudnn.so.5 libcudnn.so
$ sudo ldconfig -v