【CUDA|CUDNN】安装

every blog every motto: You can do more than you think.
https://blog.csdn.net/weixin_39190382?type=blog

0. 前言

显卡驱动安装参考之前的文章

cuda、cudnn 安装

1. cuda 安装

访问https://developer.nvidia.com/cuda-toolkit-archive

选择需要的版本:https://developer.nvidia.com/cuda-12-2-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local

python 复制代码
wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run
sudo sh cuda_12.2.0_535.54.03_linux.run

2. cudnn 安装

2.1 deb方法

访问https://developer.nvidia.com/cudnn-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_local

bash 复制代码
wget https://developer.download.nvidia.com/compute/cudnn/9.2.1/local_installers/cudnn-local-repo-ubuntu2204-9.2.1_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2204-9.2.1_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2204-9.2.1/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cudnn

对于cuda12

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

如果是cuda11

bash 复制代码
sudo apt-get -y install cudnn-cuda-11

2.2 tar方法

访问https://developer.nvidia.com/rdp/cudnn-archive

根据需求选择:

如下图所示,也可以下载deb,用deb安装

复制文件:

bash 复制代码
cd \Downloads # 更改文件夹,下载后默认地址
tar -xvf cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz # 解压
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include # 复制
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

查看cudnn版本命令:

bash 复制代码
cat /usr/include/cudnn_version.h |grep CUDNN_MAJOR -A 2

参考

  1. https://www.cnblogs.com/fireblackman/p/18202719
  2. https://cloud.tencent.com/developer/article/1726582
  3. https://blog.csdn.net/qq_32033383/article/details/135015041
  4. https://blog.csdn.net/qq_44961869/article/details/115954258
  5. https://blog.csdn.net/h3c4lenovo/article/details/119003405
  6. https://blog.csdn.net/weixin_73728875/article/details/129054053
  7. https://blog.csdn.net/yingmohuanzhou/article/details/136826468
相关推荐
Serverless 社区41 分钟前
阿里云函数计算 AgentRun 全新发布,构筑智能体时代的基础设施
人工智能·阿里云·云原生·serverless·云计算
IT_陈寒1 小时前
Python开发者必看!10个高效数据处理技巧让你的Pandas代码提速300%
前端·人工智能·后端
新智元1 小时前
全球 AI 视频大战升级!「中国版 Sora」Vidu Q2 参考生月底发布,能力对标 Sora 2
人工智能·openai
新智元2 小时前
刚刚,Figure 03 惊天登场!四年狂造 10 万台,人类保姆集体失业
人工智能·openai
万猫学社2 小时前
我们为什么需要Agent?
人工智能
共绩算力2 小时前
OpenAI Whisper 语音识别模型:技术与应用全面分析
人工智能·whisper·语音识别·共绩算力
工藤学编程2 小时前
零基础学AI大模型之Stream流式输出实战
人工智能
不良人龍木木3 小时前
机器学习-常用库
人工智能·机器学习
罗橙7号3 小时前
【pyTorch】关于PyTorch的高级索引机制理解
人工智能·pytorch·python