← 返回文档中心
镜像与环境

《预置 PyTorch / TensorFlow 深度学习环境一键启动与 Docker 镜像导入》

避免客户手动搭建 CUDA 环境耗时

一、预置镜像的选择

  • 框架镜像:预装 PyTorch / TensorFlow 常用版本,CUDA 与驱动已匹配,开机即用
  • 基础镜像:仅含 CUDA 运行时,适合自行安装框架与依赖
  • 使用预置镜像可跳过驱动 / CUDA / 框架的版本匹配工作,直接进入业务代码

二、自定义环境与镜像导入

  • 在预置镜像上用 pip / conda 安装依赖后,保存为自定义镜像供团队复用
  • Docker 镜像可按导出(docker save)→ 上传 → 导入流程在平台使用
  • 依赖建议锁定版本(requirements.txt / environment.yml),保证环境可复现

三、常见版本问题

  • CUDA 与驱动不匹配:先用 nvidia-smi 确认宿主驱动支持上限,再选对应 CUDA 版本的镜像
  • 框架与 CUDA:PyTorch / TensorFlow 官方发行包自带 CUDA 运行时,通常只需驱动满足下限
  • 驱动类报错可参考《GPU 实例 CUDA 驱动异常与 NVLink 通信延迟常见诊断步骤》
说明

可用镜像清单以控制台实际列表为准;特殊环境需求可联系技术支持定制。

需要进一步帮助?

把节点标识、报错截图与复现步骤发给我们,技术团队会更快定位。