ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
NVIDIA
NVIDIA
PyTorch
Container
NVIDIA
NVIDIA
PyTorch

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.

NVIDIA AI Enterprise Supported
LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.ref=8630bc1f3ddd81cf3d25244237386eef5517c48f
03/25/2020 3:37 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_REF
03/25/2020 3:37 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=11122848
03/25/2020 3:37 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=11122848
03/25/2020 3:37 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID
03/25/2020 3:37 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NCCL_P2P_LEVEL=PHB
03/25/2020 3:37 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
PATH=/opt/conda/bin:/opt/cmake-3.14.6-Linux-x86_64/bin/:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/opt/tensorrt/bin
03/25/2020 3:37 AM UTC
sha256:fe830d24a0daf11add8bf7bca645626ddbb5c9e9a14cc3e2fe44e59eb145cf4dRUN
NVIDIA_PYTORCH_VERSION=20.03 PYTORCH_BUILD_VERSION=1.5.0a0+8f84ded PYVER=3.6 BASE=0 DEVEL=0 SAMPLES=0 PYTHON=1 /nvidia/build-scripts/installTRT.sh &&
  mv /usr/src/tensorrt /opt &&
  ln -s /opt/tensorrt /usr/src/tensorrt
03/25/2020 3:37 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
/usr/local/bin/nvidia_entrypoint.sh
03/25/2020 3:37 AM UTC
sha256:b962e5b89d31a1c1f8fc50d848fba6850d0978747858618d6358844a955ecc80COPY
nvidia_entrypoint.sh /usr/local/bin
03/25/2020 3:37 AM UTC
...