环境要求

CPU

鲲鹏920处理器

GPU

NVIDIA Tesla A100

bazel

3.7.2

https://github.com/bazelbuild/bazel/releases/download/0.26.0/bazel-0.26.0-dist.zip

GCC

9.3.0

https://ftp.gnu.org/gnu/gcc/gcc-9.3.0/

TensorFlow

2.4.3

https://github.com/tensorflow/tensorflow/archive/v2.4.3.tar.gz

NVIDIA V100驱动

11.2

https://www.nvidia.cn/Download/index.aspx?lang=cn

NVIDIA CUDA组件

11.4

https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux_sbsa.run

jaxlib

0.17.0

https://github.com/google/jax/archive/refs/tags/jaxlib-v0.1.70.tar.gz

Python

3.8

https://www.python.org/downloads/

Miniconda

3

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2022.10-Linux-aarch64.sh

CentOS

8.2

https://www.centos.org/download/

Kernel

4.18.0-193.el8.aarch64

https://www.centos.org/download/