Ubuntu20.04 及深度学习环境anaconda、cuda、cudnn、pytorch、paddle2.3安装记录

学习目标:

Ubuntu20.04下装好torch、paddle深度学习环境。

选择的版本环境是 :最新的nvidia驱动、cuda 11.1 、cudnn v8.1.1,下面会说为啥这么选。

学习内容:

1. Ubuntu20.04仓库换源

本节参考Ubuntu 20.04 Linux更换源教程


2. Anaconda安装以及conda、pip换国内源

这个步骤可以放在后面进行

2.1 Anaconda安装

详情参考Anaconda安装,具体不再列出

要注意的一点是,anaconda安装完成后,要先关闭那个命令行,再重新打开一个命令行,才可以进行conda操作

2.2 conda、pip换国内源


3. Nvidia驱动安装

3.1 安装步骤

  1. 进入nvidia官网,下 载 NVIDIA 驱 动 安 装 包 ( .run 格 式 ),型号别选错了
  1. 然后你自己找个地方建立一个文件夹,把这个文件放进去,看着不乱。你的可能不叫这个名字,因为我给它改过名了

  2. 禁用nouveau

  3. 安装驱动

步骤中,遇到32bit的选项,选no,其他都yes(PS我踩的坑:下的nvidia驱动版本太低,默认的gcc版本是9.4.0,那个驱动要gcc 9.3.0才能装。千万别折腾gcc,会疯的!!!换个高版本驱动,最新的,就不会出问题)

3.2 说明

这里简单说明一下驱动、cuda toolkits 、cudnn的一个关系,建议你看一下,虽然我也是外行

nvidia驱动版本可以通过 nvidia-smi命令查看,我们可以看到驱动版本是515.48.07,这代表,他最高支持的CUDA版本是11.7,这是这个驱动支持CUDA工具包版本的上限。现在我想要安装Paddle深度学习框架

因此我根据自己的配置,要选CUDA 11.1工具包以及cuDNN v8.1.1工具包,所以这里不要瞎配置

即使是Torch也是有要求的,下面是11.3,其实也有11.1的,找找历史版本就行。为了保证这两个框架我都能装上,选择了以下配置(30系显卡,一定要选择CUDA 11以上的版本)

4. CUDA 11.1 安装

进入官网,选择11.1.1版本

选择离线runfile安装,下面有命令

只选第二个就行,啥都不选会报错。安装不上

安装成功,尝试nvcc -V,看到cuda 11.1就算成功了

5. CUDNN v8.1.1安装

进入官网,选择版本cudnn8.1.1 cuda11.1

下好,解压文件夹,执行命令,其实就完成了

6. Torch与Paddle安装

创建环境

按照官网给定信息,进行安装




安装paddle2.3会有问题protobuf package to 3.20.x or lower.,按照提示解决即可

对应解决方案

最后: 本期文参考 七月的和弦 博主

相关推荐
字节跳动数据库13 小时前
文章分享——相似函数处理方法
人工智能·后端·程序员
Bigfish_coding13 小时前
前端转agent-【python】-12 LangChain 入门实战:RAG + LCEL 链式调用
人工智能
程序员cxuan14 小时前
读懂 Claude Code 架构分析系列,第一篇,开始!
人工智能·后端·架构
饼干哥哥14 小时前
扣子3.0测评:我让 Codex 和 Claude Code 住同一个桌面,结果它们打架了!
人工智能·开源·代码规范
Token炼金师15 小时前
IP-Adapter:解耦交叉注意力如何让扩散模型看见图像
人工智能
Bigfish_coding15 小时前
前端转agent-【python】-11 LangGraph 高级特性:时间旅行与人工介入
人工智能
Token炼金师15 小时前
从safetensors到像素:ComfyUI Checkpoint加载机制的底层拆解
人工智能
AI闲人15 小时前
AI 写代码越来越快,为什么 Code Review 反而更慢了?
人工智能·code review·ai 编程
武子康15 小时前
调查研究-202 SGLang 深度解析:为什么大模型推理框架不只是“把模型跑起来“
人工智能·openai·agent
我是大卫15 小时前
Trae 读取 agents.md 并驱动 AI 完整底层原理
人工智能