CUDA环境安装终极指南——Linux（其它系统也一样）

文章目录

前言

不用看其它文章了，这篇文章保你不踩任何坑，安装方法简单快速

检查驱动

检查驱动是否安装，输入以下命令

nvidia-smi

如果驱动已经安装，则可跳过此步，否则，需要安装驱动。我这里装的是12.6版本

NVIDIA 驱动安装链接：https://developer.nvidia.com/cuda-12-6-0-download-archive

根据自己服务器的情况选好后，往下拉可以看到如下命令

按照步骤安装即可

安装完成后，命令行输入： nvidia-smi 能显示GPU的显存信息，则安装成功。

配置nvcc

此时，命令行输入nvcc -V，应该会显示没有该命令。

我们只需要再配置下nvcc，就可以了，在命令行输入以下命令：

vim ~/.bashrc
exportLD_LIBRARY_PATH=/usr/local/cuda/lib
exportPATH=$PATH:/usr/local/cuda/bin
source ~/.bashrc

如果一切顺利，在命令行中输入

nvcc -V

会显示：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c)2005-2024 NVIDIA Corporation
Built on Wed_Aug_14_10:10:22_PDT_2024
Cuda compilation tools, release 12.6, V12.6.68
Build cuda_12.6.r12.6/compiler.34714021_0

安装cudnn

sudoapt-get update
sudoapt-get -y install cudnn9-cuda-12

cuda11 可参阅：

Installing cuDNN on Linux

完活

The end