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

相关推荐
鬓戈9 分钟前
tmux 跑 omp / pi 等 AI coding agent:无人值守后台执行 + 随时回看操作
人工智能
Darren&Joe13 分钟前
多媒体与代码:AI Prompting 入门到高阶
人工智能
zhangfeng113323 分钟前
Ag / Cu / Au / Ru / Ir / Rh银 铜 金 钌铱 铑 纳米尺度下的 导电性 + 扩散/迁移缺点
人工智能·华为·ai编程·npu
10年前端老司机27 分钟前
实战分享:基于 PyMuPDF+Qwen-VL 实现图文兼容的 PDF RAG 方案
人工智能·python·agent
YOLO数据集集合29 分钟前
红外缺陷检测数据集 | 红外检测 缺陷识别 裂缝检测 渗漏检测 目标检测9145期
人工智能·目标检测·机器学习·计算机视觉·目标跟踪·智慧城市
2601_9628857233 分钟前
如何用 Python 回测定投策略(定期定额 DCA)?(和一次性买入对比)
开发语言·人工智能·python
蜗牛互联网33 分钟前
Python接入Gemini 3.8 Flash实现票据视觉抽取与规则校验
java·javascript·网络·人工智能·python
智圣新创0142 分钟前
锚定地方应用型高校数据资产盘活目标 一表通落地效能核验全维度高频实操答疑
人工智能
愤怒火龙果1 小时前
观察01 十字路口堵车,红绿灯能不能“看车下菜“?
人工智能·智慧城市
小鱼~~1 小时前
Recall@K、Precision@K、MRR、nDCG@K简介
人工智能·算法·机器学习