主要内容

本页采用了机器翻译。点击此处可查看英文原文。

MATLAB 深度学习容器在 Docker Hub 上

通过在 MATLAB® 深度学习容器中训练神经网络,加速您的深度学习应用程序。该容器可在 Docker Hub 上获取,专为充分利用高性能 NVIDIA® GPU 而设计。MATLAB 深度学习容器为在 AWS® 或 Microsoft® Azure® 等云环境中使用 MATLAB 进行深度学习工作流提供了简单灵活的解决方案。有关容器的更多信息,请参阅什么是容器?

内容

MATLAB 深度学习容器包含:

  • 一个 Ubuntu® 基础镜像。

  • MATLAB 及以下工具箱:

    • Computer Vision Toolbox™

    • Deep Learning Toolbox™

    • GPU Coder™

    • Image Processing Toolbox™

    • MATLAB Coder™

    • Parallel Computing Toolbox™

    • Signal Processing Toolbox™

    • Statistics and Machine Learning Toolbox™

    • Text Analytics Toolbox™

  • 若干预训练的深度神经网络。有关详细信息,请参阅 预训练的深度神经网络 (Deep Learning Toolbox)

  • 适用于深度学习工作流的实用支持包。

  • 使用容器中的 NVIDIA GPU 所需的 GPU 驱动程序。

  • 用于与 MATLAB 桌面进行交互的软件。

您可以将来自 TensorFlow™-Keras 和 Caffe 的网络及架构导入容器,可选择是否包含层权重。您还可以将训练好的网络转换为开放神经网络交换(ONNX)模型格式。有关更多信息,请参阅 导入和导出神经网络 (Deep Learning Toolbox)ONNX 转换器

要求

要使用 MATLAB 深度学习容器,您需要:

  • 安装了 Docker® 19.03 或更高版本的主机。

  • 一个符合以下条件的 MATLAB 许可证:

    • 适用于容器中安装的所有 MathWorks® 产品。您可以在 MATLAB 云端深度学习试用版中获取 MATLAB 深度学习容器产品的试用许可证。

    • 关联到一个 MathWorks 帐户

    • 已配置为在云端使用。个人许可证和全校许可证已完成配置。其他许可证类型,请联系您的许可证管理员。您可以通过查看您的 MathWorks 帐户来识别您的许可证类型和管理员。管理员可咨询管理网络许可证

  • 若您持有并发许可证,则在运行容器时必须提供网络许可证管理器的端口号和 DNS 地址。在启动容器时,将此选项添加到 docker run 命令中:-e MLM_LICENSE_FILE=27000@MyLicenseServer

MATLAB 深度学习容器快速入门指南

本节展示一个示例,说明如何运行 MATLAB 深度学习容器,并在 Web 浏览器中启动交互式 MATLAB 会话。要获取启动 MATLAB 深度学习容器的完整命令列表(包括如何通过 VNC 客户端与 MATLAB 交互以及如何在批处理模式下使用 MATLAB),请参阅 Docker Hub 上的 MATLAB 深度学习容器镜像

拉取容器镜像

要在主机上下载 MATLAB 深度学习容器镜像,请在命令行中运行以下命令:

docker pull mathworks/matlab-deep-learning:r20XYz 

您必须将标签 r20XYz 替换为具体的 MATLAB 版本名称,例如 r2022a。下载并解压容器镜像可能需要一些时间。

运行容器

使用以下命令运行 MATLAB 深度学习容器:

docker run -it --rm -p 8888:8888 --shm-size=512M mathworks/matlab-deep-learning:r20XYz -browser

  • -it 以交互模式运行容器。

  • --rm 在完成操作后会删除容器。

  • -p 8888:8888 开放端口 8888 以供 Web 浏览器连接。

  • --shm-size=512M 将共享内存大小设置为 512 MB,这是确保 MATLAB 桌面正确运行的必要条件。

  • :r20XYz 选择了 MATLAB 深度学习容器的 R20XYz 发布版本。

  • -browser 选择通过 Web 浏览器与 MATLAB 进行交互的选项。

执行上述命令会将一个 URL 打印到终端上。要访问 MATLAB,请将网址输入 Web 浏览器。若系统提示,请输入与 MATLAB 许可证关联的 MathWorks 帐户凭据。

注意

  • 从 R2022a 版本开始发布的 Docker 镜像支持 -browser 选项。要在自定义 Docker 镜像或旧版 MATLAB Docker 镜像中通过 Web 浏览器访问 MATLAB,请参阅示例

  • 某些浏览器不支持 -browser 选项。有关详细信息,请参阅云解决方案浏览器需求

其他信息

默认情况下,容器无法访问主机系统的硬件资源。要使容器能够访问主机系统的 GPU,请使用 docker run 命令的 --gpus 标志。要使容器能够访问主机系统上的所有 GPU,请将此标志设置为 all。例如,执行此命令将启动一个名为 MATLAB 的容器,该容器可访问主机系统中的所有 GPU:

docker run --gpus all -it --rm --shm-size=512M mathworks/matlab-deep-learning:r2022a

有关详细信息,请参阅在容器中使用 GPU

要查看启动容器时可用的完整选项和环境变量列表,请使用 -help 标志运行容器:

docker run -it --rm mathworks/matlab-deep-learning:r20XYz -help

有关使用环境变量配置 MathWorks 容器的更多信息,请参阅 配置容器

另请参阅

主题

外部网站