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

相关推荐
爱学习的张大1 分钟前
深度学习中稀疏专家模型研究综述 A REVIEW OF SPARSE EXPERT MODELS IN DEEP LEARNING
人工智能·深度学习
爱打代码的小林7 分钟前
CNN 卷积神经网络 (MNIST 手写数字数据集的分类)
人工智能·分类·cnn
川西胖墩墩8 分钟前
游戏NPC的动态决策与情感模拟
人工智能
E_ICEBLUE9 分钟前
零成本实现文档智能:本地化 OCR 提取与 AI 处理全流程实战
人工智能·ocr
乾元10 分钟前
无线定位与链路质量预测——从“知道你在哪”,到“提前知道你会不会掉线”的网络服务化实践
运维·开发语言·人工智能·网络协议·重构·信息与通信
MistaCloud10 分钟前
Pytorch深入浅出(十五)之GPU加速与设备管理
人工智能·pytorch·python·深度学习
源于花海11 分钟前
迁移学习的第一类方法:数据分布自适应(3)——联合分布自适应
人工智能·机器学习·迁移学习·联合分布自适应
梁辰兴12 分钟前
中国信通院发布《人工智能安全治理研究报告(2025年)》,AI安全攻防为何“易攻难守“?
人工智能·安全·ai·ai安全·梁辰兴·人工智能安全治理·中国信通院
Suahi20 分钟前
【HuggingFace LLM】规范化与预分词(BPE、WordPiece以及Unigram)
大数据·人工智能
元智启24 分钟前
企业 AI 应用进入 “能力解耦时代”:模块化重构 AI 落地新范式
大数据·人工智能·重构