install CUDA CUDNN

最新推荐文章于 2024-10-25 23:46:10 发布

LinFengOnlyOne

最新推荐文章于 2024-10-25 23:46:10 发布

阅读量324

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/LinFengOnlyOne/article/details/99942624

本文详细介绍如何在Linux系统上安装NVIDIA GPU驱动、CUDA和CUDNN，包括卸载CUDA的步骤。首先，通过nvidia-smi和lsmod检查当前GPU驱动状态，然后使用rpm包安装NVIDIA驱动。接着，通过运行.sh文件安装CUDA工具包，并修改.bashrc文件设置环境变量。最后，解压并复制CUDNN库到CUDA目录下，更新符号链接和权限。

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

#install nvidia gpu driver
nvidia-smi
lsmod | grep nouveau
sudo rpm -ivh nvidia-diag-driver-local-repo-*.x86_64.rpm

#install CUDA CUDNN
sudo rm /tmp/.X0-lock
sudo sh suda_*.run
sudo vi ~/.bashrc
#insert
export PATH = /usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH = /usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc
sudo ldconfig
nvcc --version

sudo tar -xvf cudnn-9.0-linnux-x64*.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.6.0 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig

sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

#uninstall cuda
sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl
sudo rm -r /usr/local/cuda-9.0