Ubuntu系统安装cudn与cudnn

最新推荐文章于 2024-07-25 17:01:46 发布

aoguren

最新推荐文章于 2024-07-25 17:01:46 发布

阅读量1.5k

点赞数

分类专栏：人工智能文章标签： cudn

本文链接：https://blog.csdn.net/aoguren/article/details/79960025

版权

本文详细介绍了在Ubuntu系统中安装CUDA和cuDNN的步骤。首先，通过查看显卡信息并安装合适的显卡驱动。接着，选择.run文件安装CUDA，注意避免潜在问题，并验证安装成功。然后，配置相关环境。对于cuDNN，下载适用于CUDA 9.0的v7.1.2版本，包括运行时和开发者库，完成安装后，通过TensorFlow-gpu测试确保cuDNN安装无误。

摘要由CSDN通过智能技术生成

一、cudn安装

1、显卡驱动安装

apt-get update
apt-get install cuda-drivers

查看显卡信息

nvidia-smi

2、cudn安装

注意：选择runfile，其他的有些有坑。

安装命令：

chmod +x cuda_9.0.176_384.81_linux.run
sudo ./cuda_9.0.176_384.81_linux.run --override

安装过程的选择：

Do you accept the previously read EULA?
accept/decline/quit: accept 

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
(y)es/(n)o/(q)uit: n （因为之前安装过显卡驱动，因而选择n）

Install the CUDA 9.0 Toolkit?
(y)es/(n)o/(q)uit: y

Enter Toolkit Location
 [ default is /usr/local/cuda-9.0 ]:

查看是否安装成功