openeuler/pytorch Docker Image

快速参考

• 官方PyTorch Docker镜像。

• 维护者：https://gitee.com/openeuler/cloudnative。

• 获取帮助：https://gitee.com/openeuler/cloudnative、https://gitee.com/openeuler/community。

PyTorch | openEuler

当前PyTorch Docker镜像基于https://repo.openeuler.org/构建。本仓库可免费使用，且无每用户速率限制。

PyTorch是一个Python包，提供两个高级功能：

• 张量计算（类似NumPy），支持强大的GPU加速
• 基于磁带式自动微分系统构建的深度神经网络

必要时，您可以重用NumPy、SciPy和Cython等常用Python包来扩展PyTorch功能。

更多信息请访问https://pytorch.org/docs/stable/index.html。

支持的标签及对应Dockerfile链接

每个pytorch镜像的标签由完整软件栈版本组成，详情如下：

使用方法

用户可根据需求选择相应的{Tag}和容器启动选项。

• 从Docker拉取openeuler/pytorch镜像

  bash
  docker pull docker.xuanyuan.run/openeuler/pytorch:{Tag}
  

• 启动PyTorch实例

  bash
  docker run \
      --name my-pytorch \
      --device /dev/davinci1 \
      --device /dev/davinci_manager \
      --device /dev/devmm_svm \
      --device /dev/hisi_hdc \
      -v /usr/local/dcmi:/usr/local/dcmi \
      -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
      -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
      -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
      -v /etc/ascend_install.info:/etc/ascend_install.info \
      -it docker.xuanyuan.run/openeuler/pytorch:{Tag} bash
  

• 容器启动选项说明

  选项	描述
  --name my-pytorch	为容器命名为my-pytorch。
  --device /dev/davinciX	NPU设备，其中X为芯片物理ID号，例如davinci1。
  --device /dev/davinci_manager	Davinci相关管理设备。
  --device /dev/devmm_svm	内存管理相关设备。
  --device /dev/hisi_hdc	HDC相关管理设备。
  -v /usr/local/dcmi:/usr/local/dcmi	将主机的DCMI .so和接口文件目录/usr/local/dcmi挂载到容器，需根据实际情况修改。
  -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi	将主机npu-smi工具"/usr/local/bin/npu-smi"挂载到容器，需根据实际情况修改。
  -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/	将主机目录/usr/local/Ascend/driver/lib64/挂载到容器，需根据驱动.so文件实际路径修改。
  -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info	将主机版本信息文件/usr/local/Ascend/driver/version.info挂载到容器，需根据实际情况修改。
  -v /etc/ascend_install.info:/etc/ascend_install.info	将主机安装信息文件/etc/ascend_install.info挂载到容器。
  -it	以交互模式启动容器并分配终端（bash）。
  openeuler/pytorch:{Tag}	指定要运行的Docker镜像，将{Tag}替换为所需openeuler/pytorch镜像的具体版本或标签。

• 查看容器运行日志

  bash
  docker logs -f my-pytorch
  

• 获取交互式shell

  bash
  docker exec -it my-pytorch /bin/bash
  

问题与解答

如有任何问题或需使用特定功能，请在https://gitee.com/openeuler/openeuler-docker-images提交issue或pull request。