Application | Versions | Description | Multi-Node |
| PyTorch 2.0.0 | Includes:- NVIDIA Driver release 530
- Ubuntu 20.04, CUDA 12.1.0, PyTorch 2.0.0, TensorRT 8.6.1
- Python 3.8 and other key frameworks
- Pre-installed python packages:
- sklearn, pandas, numpy, seaborn, matplotlib, scipy
- SSH server for command-line access and SFTP/SCP
- Integrated JupyterLab notebook server
Based on NVIDIA NGC PyTorch 23.03
| Yes |
| PyTorch 1.14.0 | Includes:- NVIDIA Driver release 530
- Ubuntu 20.04, CUDA 12.0.1, PyTorch 1.14.0, TensorRT 8.5.3
- Python 3.8 and other key frameworks
- Pre-installed python packages:
- sklearn, pandas, numpy, seaborn, matplotlib, scipy
- SSH server for command-line access and SFTP/SCP
- Integrated JupyterLab notebook server
Based on NVIDIA NGC PyTorch 23.02 | Yes |
| RAPIDS 22.12 | Includes:- NVIDIA Driver release 530
- Ubuntu 20.04, CUDA 11.8, RAPIDS 22.12
- Python 3.10 and other key frameworks
- SSH server for command-line access and SFTP/SCP
- Integrated JupyterLab notebook server
- Integrated Dask scheduler
Based on NVIDIA NGC PyTorch 23.02 | Yes |
| TensorFlow 2.11.0 | Includes: - NVIDIA Driver release 530
- Ubuntu 20.04, CUDA 12.0.1, TensorFlow 2.11.0, TensorRT 8.5.3
- Python 3.8 and other key frameworks
- Pre-installed python packages:
- sklearn, pandas, numpy, seaborn, matplotlib, scipy
- SSH server for command-line access and SFTP/SCP
- Integrated JupyterLab notebook server
Based on NVIDIA NGC TensorFlow 23.02 | Yes |
| Apache MXNet 1.5.0 | Includes: - GPU drivers, python3 and common Linux tools
- Python packages:
- MXNet and CUDA libraries
- sklearn, pandas, numpy, seaborn, matplotlib, scipy
- SSH server for command-line access and SFTP/SCP
- Integrated JupyterLab notebook server
Based on ubuntu:22.04
| Yes |