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

文章目录


前言

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

检查驱动

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

bash 复制代码
nvidia-smi

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

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

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

按照步骤安装即可

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

配置nvcc

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

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

bash 复制代码
vim ~/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda/lib
export PATH=$PATH:/usr/local/cuda/bin
source ~/.bashrc

如果一切顺利,在命令行中输入nvcc -V会显示:

bash 复制代码
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

bash 复制代码
sudo apt-get update
sudo apt-get -y install cudnn9-cuda-12

cuda11 可参阅:

Installing cuDNN on Linux

完活

The end

相关推荐
心中有国也有家21 小时前
hccl 架构拆解:昇腾集合通信库到底在做什么?
人工智能·经验分享·笔记·分布式·算法·架构
這花開嗎1 天前
试了一圈配音网站,说说我的感受
人工智能·语音识别
w_t_y_y1 天前
AI应用demo(二)打造个人的code agent
人工智能·语音识别
Raink老师1 天前
【AI面试临阵磨枪-60】微服务下 AI 能力如何封装、网关、限流、监控
人工智能·微服务·面试
ApiHug1 天前
Mintlify、Stainless & ApiHug 在AI 时代的战略意义
人工智能
九皇叔叔1 天前
Spring-Ai-Alibaba [04] 04-llm-platform-custom-demo
java·人工智能·spring
ai生成式引擎优化技术1 天前
DLOS Kernel v0.5:从多Agent系统到AI操作系统内核的拐点
人工智能
CHEN5_021 天前
深入理解 RAG(检索增强生成):核心流程、技术选型与进阶实战
人工智能·rag
@蔓蔓喜欢你1 天前
团队协作工具:提升开发效率的利器
人工智能·ai
T.i.s1 天前
parall scan(并行扫描)通俗理解
人工智能·深度学习