ubuntu16.04安装cuda9、cudnn

最新推荐文章于 2024-08-24 11:45:15 发布

原创最新推荐文章于 2024-08-24 11:45:15 发布 · 648 阅读

0 ·

CC 4.0 BY-SA版权

本文详细介绍了在Ubuntu系统上安装NVIDIA驱动、CUDA9.0、cuDNN的步骤，包括驱动安装、环境变量配置及验证安装是否成功的方法。

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

（1）首先要安装NVIDIA显卡驱动，特别对于新电脑，再找不到nvidia-smi指令时：

打开命令窗口（ctrl+Alt+T）

sudo apt-get purge nvidia*

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

sudo apt-get update

sudo apt-get install nvidia-384 nvidia-settings (sudo apt-get install nvidia-410 nvidia-settings)

安装完毕后，运行nvidia-smi，查看生效的显卡驱动。若不显示，重启一下再输。

（2）安装CUDA

下载CUDA9.0版本（runfile），命令行输入（版本各异，使用tab去补全）：

sudo bash cuda_9.0.176_384.81_linux.run

(3)安装cuDNN

下载cuDNN for CUDA9.0（压缩包），解压后会得到cuda文件夹（注意目录不能带中文，否则解压失败）。命令行输入：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d

(4)修改path

命令行输入：

sudo gedit ~/.bashrc

打开文件后，在最后面添加：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64

export PATH=$PATH:/usr/local/cuda-9.0/bin

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

(5)命令行输入：