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

相关推荐
那个村的李富贵4 小时前
光影魔术师:CANN加速实时图像风格迁移,让每张照片秒变大师画作
人工智能·aigc·cann
腾讯云开发者5 小时前
“痛点”到“通点”!一份让 AI 真正落地产生真金白银的实战指南
人工智能
CareyWYR5 小时前
每周AI论文速递(260202-260206)
人工智能
hopsky6 小时前
大模型生成PPT的技术原理
人工智能
禁默7 小时前
打通 AI 与信号处理的“任督二脉”:Ascend SIP Boost 加速库深度实战
人工智能·信号处理·cann
心疼你的一切7 小时前
昇腾CANN实战落地:从智慧城市到AIGC,解锁五大行业AI应用的算力密码
数据仓库·人工智能·深度学习·aigc·智慧城市·cann
AI绘画哇哒哒7 小时前
【干货收藏】深度解析AI Agent框架:设计原理+主流选型+项目实操,一站式学习指南
人工智能·学习·ai·程序员·大模型·产品经理·转行
数据分析能量站7 小时前
Clawdbot(现名Moltbot)-现状分析
人工智能
那个村的李富贵7 小时前
CANN加速下的AIGC“即时翻译”:AI语音克隆与实时变声实战
人工智能·算法·aigc·cann
二十雨辰7 小时前
[python]-AI大模型
开发语言·人工智能·python