- Version 21.02.2
- Sold by NVIDIA
NVDIA's GPU-optimized distribution of PyTorch. PyTorch is a GPU accelerated tensor computational framework with a Python front end. Functionality can be easily extended with common Python libraries such as NumPy, SciPy and Cython. This image bundles NVIDIA's container for PyTorch into the NGC...
Linux/Unix, Ubuntu Ubuntu 18.04 - 64-bit Amazon Machine Image (AMI)