dsksd/pytorch

Docker镜像文档

镜像概述

本Docker镜像系列提供了包含特定版本Python、CUDA、CuDNN及PyTorch的深度学习运行环境，旨在满足对旧版本框架有依赖的应用场景，确保环境一致性和兼容性。

核心功能与特性

版本说明

• 0.4版本：基础环境，包含Python 3.6、CUDA 8、CuDNN 6及PyTorch 0.4
• 0.4a版本：在0.4版本基础上，新增AllenNLP和spaCy（英文模型）
• 0.4.1版本：在0.4a版本基础上，将PyTorch更新至0.4.1，保留AllenNLP和spaCy（英文模型）

使用场景

• 学术研究中依赖旧版本PyTorch（0.4.x）的实验复现
• 需要CUDA 8/CuDNN 6环境的深度学习模型部署
• 集成AllenNLP和spaCy进行自然语言处理任务的应用开发

使用方法

拉取镜像

根据需求选择对应版本标签拉取镜像：

bash
# 拉取0.4版本
docker pull [镜像仓库地址]:0.4

# 拉取0.4a版本
docker pull [镜像仓库地址]:0.4a

# 拉取0.4.1版本
docker pull [镜像仓库地址]:0.4.1

运行容器

以交互式方式运行容器（示例以0.4版本为例）：

bash
docker run -it --gpus all [镜像仓库地址]:0.4 /bin/bash

说明：--gpus all参数用于启用GPU支持，需确保主机已安装NVIDIA Docker运行时

验证环境

容器启动后，可通过以下命令验证环境配置：

bash
# 验证Python版本
python --version  # 应输出Python 3.6.x

# 验证PyTorch版本及GPU支持
python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"  # 0.4版本应输出0.4.0和True

# 0.4a/0.4.1版本额外验证AllenNLP和spaCy
python -c "import allennlp; import spacy; nlp = spacy.load('en'); print('AllenNLP version:', allennlp.__version__); print('spaCy version:', spacy.__version__)"

注意事项

• 该镜像基于CUDA 8，需确保主机GPU驱动支持CUDA 8
• 不同版本针对特定依赖组合优化，选择时需匹配项目依赖要求
• 若需持久化数据，可通过-v参数挂载主机目录：docker run -it -v /host/path:/container/path --gpus all [镜像仓库地址]:[版本] /bin/bash