当前位置: 首页 > article >正文

使用 PaddlePaddle 官方提供的 Docker 镜像

CUDA版本高PaddlePaddle不支持时,可以使用 PaddlePaddle 官方提供的 Docker 镜像

      • 1. 安装 Docker Desktop
        • 1.1 下载 Docker Desktop
        • 1.2 安装 Docker Desktop
        • 1.3 启用 WSL 2 或 Hyper-V
        • 1.4 启动 Docker Desktop
        • 1.5 Docker不运行解决方法
      • 2. 拉取 PaddlePaddle Docker 镜像
        • 2.1 打开终端
        • 2.2 拉取镜像
      • 3. 运行 Docker 容器
        • 3.1 运行 GPU 容器
        • 3.2 运行 CPU 容器
      • 4. 验证 PaddlePaddle 安装
      • 5. 使用 Docker 容器
        • 5.1 运行 Python 脚本
      • 6. 退出和删除容器
      • 7. 常见问题
        • 7.1 Docker 无法启动
        • 7.2 GPU 不支持

在 Windows 系统上使用 Docker 镜像运行 PaddlePaddle 或其他应用程序非常简单。以下是详细步骤:

1. 安装 Docker Desktop

Docker Desktop 是 Windows 上运行 Docker 的官方工具。以下是安装步骤:

1.1 下载 Docker Desktop

访问 Docker 官方网站。

下载适用于 Windows 的 Docker Desktop 安装程序。

1.2 安装 Docker Desktop

1、运行下载的安装程序。

2、按照提示完成安装。

3、安装完成后,重启电脑。

1.3 启用 WSL 2 或 Hyper-V

Docker Desktop 需要 WSL 2(Windows Subsystem for Linux 2)或 Hyper-V 支持。

#WSL 2(推荐):

1、打开 PowerShell(管理员权限)。

2、运行以下命令启用 WSL 2:

wsl --install

3、安装完成后,重启电脑。

#Hyper-V:

1、打开“控制面板” -> “程序” -> “启用或关闭 Windows 功能”。

2、勾选“Hyper-V”并点击“确定”。

3、重启电脑。

1.4 启动 Docker Desktop

1、在开始菜单中搜索并启动 Docker Desktop。

2、等待 Docker 启动(右下角系统托盘中的 Docker 图标变为绿色)。

1.5 Docker不运行解决方法

如果界面显示Engine stopped,则卸载Docker Desktop,在安装 Docker Desktop 之前,先安装 WSL 2 并配置至少一个 Linux 发行版(如 Ubuntu)。 参考以下两篇文章
全网最全Win10/11系统下WSL2+Ubuntu20.04的全流程安装指南(两种支持安装至 D 盘方式)
一篇就够!Windows上Docker Desktop安装 + 汉化完整指南(包含解决wsl更新失败方案)

2. 拉取 PaddlePaddle Docker 镜像

PaddlePaddle 提供了预配置的 Docker 镜像,支持 GPU 和 CPU。

2.1 打开终端

打开 PowerShell 或 Windows 终端。

2.2 拉取镜像

运行以下命令拉取 PaddlePaddle 镜像(以 GPU 版本为例):

docker pull paddlepaddle/paddle:2.5.2-gpu-cuda11.7-cudnn8

3. 运行 Docker 容器

拉取镜像后,可以通过以下命令运行容器。

3.1 运行 GPU 容器

如果你的系统支持 GPU,可以使用以下命令运行容器:

docker run --gpus all -it paddlepaddle/paddle:2.5.2-gpu-cuda11.7-cudnn8 /bin/bash
3.2 运行 CPU 容器

如果你的系统不支持 GPU,可以使用以下命令运行 CPU 容器:

docker run -it paddlepaddle/paddle:2.5.2 /bin/bash

4. 验证 PaddlePaddle 安装

在容器中运行以下命令,验证 PaddlePaddle 是否正确安装:

python -c "import paddle; print(paddle.device.is_compiled_with_cuda())"

如果输出 True,说明 GPU 支持已启用。

如果输出 False,说明当前运行的是 CPU 版本。

5. 使用 Docker 容器

在容器中,你可以像在本地环境中一样运行 Python 脚本或 Jupyter Notebook。

5.1 运行 Python 脚本

将本地脚本挂载到容器中并运行:

docker run --gpus all -v C:/path/to/your/script:/workspace -it paddlepaddle/paddle:2.5.2-gpu-cuda11.7-cudnn8 /bin/bash
cd /workspace
python your_script.py

5.2 运行 Jupyter Notebook
启动 Jupyter Notebook 服务:

docker run --gpus all -p 8888:8888 -v C:/path/to/your/notebooks:/workspace paddlepaddle/paddle:2.5.2-gpu-cuda11.7-cudnn8 jupyter notebook --ip=0.0.0.0 --allow-root

然后在浏览器中访问 http://localhost:8888。

6. 退出和删除容器

退出容器:在容器中运行 exit。

删除容器:使用以下命令删除已停止的容器:

docker rm <container_id>

7. 常见问题

7.1 Docker 无法启动

确保 WSL 2 或 Hyper-V 已启用。

确保 BIOS 中启用了虚拟化技术(VT-x/AMD-V)。

7.2 GPU 不支持

确保已安装 NVIDIA 驱动程序。

确保已安装 NVIDIA Container Toolkit。


http://www.kler.cn/a/587615.html

相关文章:

  • 超声重建,3D重建 超声三维重建,三维可视化平台 UR 3D Reconstruction
  • PCDN 与边缘计算的结合​
  • PyTorch 深度学习实战(14):Deep Deterministic Policy Gradient (DDPG) 算法
  • Linux 命令学习记录
  • 数位小游戏
  • 聊聊 Redis 的一些有趣的特性(上)
  • ubuntu系统下添加pycharm到快捷启动栏方法
  • 【Qt】qApp简单介绍
  • ENSP实验案例-企业局域网搭建实施设计方案(计算机毕业设计作品)
  • html-to-image的使用及图片变形和无图问题修复
  • Hive SQL 精进系列:SUBSTR 函数的多样用法
  • 基于Python的天气预报数据可视化分析系统-Flask+html
  • 仿最美博客POETIZE(简易版)
  • 【go】Go 语言中 errors.Is 和 errors.As 的区别
  • LeetCode-跳跃游戏 II
  • std::invoke详解
  • 【C++】string类
  • <rust><tauri><GUI>基于tauri和rust,编写一个二维码生成器
  • 【鸿蒙开发】Hi3861学习笔记-Visual Studio Code安装(New)
  • 在线招聘小程序:AI简历筛选与精准职位推荐服务