Skip to content

概述


基础系统环境

  • 操作系统:Ubuntu
  • 主要版本:22.04 LTS
  • 可选版本:18.04 LTS / 20.04 LTS

预装框架服务

平台提供开箱即用的深度学习环境,预装以下主流框架:

框架类型版本范围Python支持CUDA版本
PyTorch1.5.1~2.7.03.8~3.1210.1~12.6
TensorFlow2.7.0~2.9.33.811.2~11.4
JAX0.4.3012.1-

环境配置方案

  1. 【推荐】预装镜像
    ✅ 自动配置完整环境
    示例组合

    • PyTorch 2.1.0
    • Python 3.10
    • CUDA 12.1
  2. 【定制】基于CUDA版本

    • 查询框架CUDA要求
    • 选择对应Miniconda镜像
    • 手动安装目标框架
  3. 【高级】完全自定义

    • 选择Miniconda基础镜像
    • 自行安装:
      • Python (3.7-3.12)
      • CUDA工具包 (9.0-12.6)
      • 所需框架

💡 特别提示

  1. PyTorch安装建议
bash
pip install torch torchvision --index-url https://pypi.tuna.tsinghua.edu.cn/simple

🚫 危险操作预警 避免使用conda安装(可能导致误装CPU版本)

  1. 平台优势:
  • 所有CUDA环境均包含.h头文件
  • 支持二次编译需求
  • 已配置国内镜像源加速下载