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.,按照提示解决即可

对应解决方案

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

相关推荐
攻城狮7号1 天前
Meta开源SAM 3D,如何教会 AI “想象”三维世界
人工智能·开源大模型·sam 3d·2d变3d·meta大模型·3d 视觉ai
三七互娱后端团队1 天前
告别“玄学”调参:DSPy 框架入门,让 AI 自动优化 AI 的提示词
人工智能·后端
三七互娱后端团队1 天前
别再只用 Vector Search 了:手把手教你落地 GraphRAG(图谱增强检索)
人工智能·后端
Predestination王瀞潞1 天前
Windows环境下Pytorch的配置
人工智能·pytorch·python
Christo31 天前
AAAI-2024《Multi-Class Support Vector Machine with Maximizing Minimum Margin》
人工智能·算法·机器学习·支持向量机·数据挖掘
千里飞刀客1 天前
aruco位姿检测
人工智能·opencv·计算机视觉
浪子不回头4151 天前
AI机考-Transformers
人工智能
BAOYUCompany1 天前
暴雨AI服务器点燃AGI蓝海市场
人工智能
枯木逢秋࿐1 天前
深度学习常用模型
深度学习
神一样的老师1 天前
Google学术搜索实验室:自然语言检索新体验
人工智能