概要

平台所有镜像的系统版本为Ubuntu20.04
平台已经内置了以下框架及版本的镜像,使用该镜像的实例就会自带相应框架软件。如果以下自带的框架版本或Python版本不满足需求,请继续看下方配置其他版本的框架或Python方法。
框架框架版本Python版本CUDA版本
PyTorch1.5.13.8.1010.1
PyTorch1.6.03.8.1010.1
PyTorch1.7.03.811.0
PyTorch1.8.13.811.1
PyTorch1.9.03.811.1
PyTorch1.10.03.811.3
PyTorch1.11.03.811.3
TensorFlow2.9.03.811.2
TensorFlow2.13.03.1011.8
TensorFlow2.12.03.1011.8
TensorFlow
2.15.03.810.0
PaddlePaddle2.2.03.811.2
Minicondaconda33.811.3
Minicondaconda33.811.6
Minicondaconda33.811.8
TensorRT 
8.5.13.811.8
TensorRT
8.6.13.811.8
清华Jittor1.3.13.711.3



安装其他版本的Python: 参考

安装其他版本的CUDA: 参考

安装PyTorch: 参考

安装TensorFlow: 参考

推荐的使用姿势

  1. 首先平台镜像中有没有您需要的Torch、TensorFlow等框架的相应版本,如果有首选平台内置的镜像
  2. 如果平台中没有合适的Torch、TensorFlow等框架版本,那么查询自己的框架需要什么CUDA版本,比如PyTorch=1.9.0需要CUDA=11.1,那么可以选择Miniconda/CUDA=11.1的平台镜像,然后在镜像内安装自己需要的框架,免去安装cudatoolkit的麻烦。(平台内置的CUDA均带.h头文件,如有二次编译代码的需求更方便)
  3. 如果以上条件都不满足,则可随便挑选一个Miniconda镜像,在开机后自行安装相关框架、CUDA、甚至其他版本的Python。
栏目
问题反馈