Skip to content
共绩算力文档中心

概要

平台已内置以下框架及版本的镜像,使用该基础镜像将预装对应的框架与环境。若下列版本不符合您的需求,请参考后续配置其他版本框架或 Python 的方法。

框架

框架版本

Python 版本

CUDA 版本

PyTorch

2.7.1

3.12 (Ubuntu 22.04)

12.8

PyTorch

2.7.1

3.12 (Ubuntu 20.04)

12.8

PyTorch

2.7.0

3.12 (Ubuntu 22.04)

12.4

PyTorch

2.6.0

3.12 (Ubuntu 20.04)

12.6

PyTorch

2.5.1

3.12 (Ubuntu 22.04)

12.4

PyTorch

2.4.1

3.12 (Ubuntu 20.04)

12.4

TensorFlow

2.19

3.12 (Ubuntu 20.04)

12.4

TensorFlow

2.16

3.12 (Ubuntu 22.04)

12.2

TensorFlow

2.15

3.11 (Ubuntu 20.04)

12.2

TensorFlow

2.11

3.10 (Ubuntu 20.04)

12.2

Paddle

3.10

3.12 (Ubuntu 20.04)

11.2

Ubuntu

22.04

3.10

Ubuntu

24.04

3.12

推荐的使用姿势

  1. 首先平台镜像中有没有您需要的 Torch、TensorFlow 等框架的相应版本,如果有首选平台内置的镜像。
  2. 如果平台中没有合适的 Torch、TensorFlow 等框架版本,那么查询自己的框架需要什么 CUDA 版本,比如 PyTorch=1.9.0 需要 CUDA=11.1,那么可以选择 Ubuntu 的平台镜像,然后在镜像内安装自己需要的框架。