Component | Requirement |
---|---|
Operating System |
|
CPU | x86-64 |
GPU | NVIDIA GPU supports CUDA® 10.2 or higher, such as NVIDIA P100, V100, A100 |
CUDA Driver | Driver compatible to CUDA version, NVIDIA-384 or higher for CUDA10.2, NVIDIA-450 or higher for CUDA 11.0 |
Docker version | 19.03 or higher |
Note: Only vai_p_pytorch and vai_p_tensorflow2 tools
support the NVIDIA A100 Tensor Core GPU.