logo
0
0
WeChat Login

PyTorch 开发环境 Docker 镜像

该项目提供了一个预配置的 Docker 环境,专为 PyTorch 深度学习开发而设计。它集成了 Web VSCode (code-server)、增强型 Zsh 终端以及最新版本的 PyTorch。

主要特性

  • PyTorch 2.9.0: 预装了 PyTorch、TorchVision 和 TorchAudio,支持 CUDA 13.0。
  • Web VSCode (code-server): 内置浏览器版 VS Code,支持远程开发,并预装了 Coding Copilot 插件。
  • 增强型终端 (Zsh):
    • 预装了 Oh My Zsh。
    • 启用了 zsh-autosuggestions (自动建议) 和 zsh-syntax-highlighting (语法高亮) 插件。
    • 优化的 settings.json 配置,提升开发体验。
  • 自动化脚本: 包含用于自动配置 Zsh 插件和环境变量的脚本。

目录结构

/ ├── Dockerfile # 镜像构建文件 ├── settings.json # VS Code 配置文件 └── scripts/ # 辅助配置脚本 ├── add-zsh-plugins.sh # 自动添加 Zsh 插件 └── set-zsh-env.sh # 设置环境变量(如 VS Code Server 路径)

配置详情

  • Python 环境: 位于 /opt/venv,已加入系统 PATH。
  • VS Code 配置: 默认启用了自动保存 (onFocusChange),并优化了字体和终端显示。
  • Zsh 配置: 默认 Shell 已切换为 Zsh,并集成了常用的开发辅助插件。

该项目旨在为开发者提供一个“开箱即用”的稳定深度学习开发环境。

About

No description, topics, or website provided.
Pytorchrepo-named
Language
Dockerfile55%
Shell45.1%