概要
平台已内置以下框架及版本的镜像,使用该基础镜像将预装对应的框架与环境。若下列版本不符合您的需求,请参考后续配置其他版本框架或 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 | 无 |
推荐的使用姿势:
- 首先平台镜像中有没有您需要的 Torch、TensorFlow 等框架的相应版本,如果有首选平台内置的镜像。
- 如果平台中没有合适的 Torch、TensorFlow 等框架版本,那么查询自己的框架需要什么 CUDA 版本,比如 PyTorch=1.9.0 需要 CUDA=11.1,那么可以选择 Ubuntu 的平台镜像,然后在镜像内安装自己需要的框架。