| 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 the vai_p_pytorch and vai_p_tensorflow2 tools
support the NVIDIA A100 Tensor Core GPU.