PyTorch | NVIDIA NGC
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorchPyTorch | NVIDIA NGC Description PyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels. Publisher Facebook Latest Tag 21.11-py3 Modified