seemeai/fastai Docker 镜像 - 轩辕镜像 | Docker 镜像高效稳定拉取服务

fast.ai Docker容器

镜像概述和主要用途

该Docker容器旨在帮助用户轻松开始使用fast.ai框架（GitHub仓库：[***] GPU的Linux机器上使用。

核心功能和特性

• 支持fast.ai框架的快速部署和使用
• 集成PyTorch深度学习框架
• 通过NVIDIA GPU加速（需配合nvidia-docker）
• 数据持久化：通过卷挂载保存/root/.fastai和/root/.cache目录数据
• 可配置容器内存大小（通过--shm-size参数）

使用场景和适用范围

• 适用于使用fast.ai框架进行深度学习模型开发、训练和推理的场景
• 适合希望快速搭建fast.ai环境的开发者和研究人员
• 需运行在支持NVIDIA GPU的Linux系统上

详细使用方法和配置说明

前置条件

1. 安装Docker
   参考官方文档：[***]

2. 安装nvidia-docker
   参考官方文档：[***]

运行容器

使用以下命令启动容器：

bash
docker run -it --gpus all --shm-size=32g -p 8888:8888 -v $HOME/.fastai:/root/.fastai -v $HOME/.cache:/root/.cache seemeai/fastai:tagname

参数说明：

• --gpus all：允许容器使用所有GPU资源
• --shm-size=32g：设置容器可使用的内存大小（根据需求调整）
• -p 8888:8888：端口映射，将容器的8888端口映射到主机的8888端口（适用于Jupyter Notebook等服务）
• 卷挂载：
  • -v $HOME/.fastai:/root/.fastai：持久化fast.ai数据
  • -v $HOME/.cache:/root/.cache：持久化缓存数据
• seemeai/fastai:tagname：容器镜像名及标签（需替换tagname为实际标签）

标签说明

命名规则

标签命名格式为：pt-%PYTORCH_VERSION-fa-%FASTAI_VERSION-%BUILD_VERSION，包含以下组件：

• pt-%PYTORCH_VERSION：PyTorch版本
• fa-%FASTAI_VERSION：fastai版本
• %BUILD_VERSION：构建版本

示例：标签pt-1.11.0-fa-2.7.2-1.1.0表示：

• PyTorch版本：1.11.0
• fastai版本：2.7.2
• 构建版本：1.1.0

torchvision版本对应

每个PyTorch版本对应特定的torchvision版本，如下表所示：

更多关于torchvision的信息：[***]

基础容器

当前基础容器为：pytorch/pytorch:$torch_version-cuda11.3-cudnn8-runtime，其中torch_version支持1.10.0或1.11.0。

许可证

MIT许可证

联系方式

---