CUDA环境安装终极指南——Linux(其它系统也一样)
文章目录
- 前言
- 检查驱动
- 配置nvcc
- 安装cudnn
- 完活
前言
不用看其它文章了,这篇文章保你不踩任何坑,安装方法简单快速
检查驱动
检查驱动是否安装,输入以下命令
nvidia-smi
如果驱动已经安装,则可跳过此步,否则,需要安装驱动。我这里装的是12.6版本
NVIDIA 驱动安装链接:https://developer.nvidia.com/cuda-12-6-0-download-archive
根据自己服务器的情况选好后,往下拉可以看到如下命令
按照步骤安装即可
安装完成后,命令行输入: nvidia-smi 能显示GPU的显存信息,则安装成功。
配置nvcc
此时,命令行输入nvcc -V,应该会显示没有该命令。
我们只需要再配置下nvcc,就可以了,在命令行输入以下命令:
vim ~/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda/lib
export PATH=$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
sudo apt-get update
sudo apt-get -y install cudnn9-cuda-12
cuda11 可参阅:
Installing cuDNN on Linux
完活
The end