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
c13989922143b4f8353c4b869027916b00b855a01f63b3a67a7a5c387b3c0eb2CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
12/05/2023 10:36 PM UTC
18d9d42e9d17d740fd2b7a2e8dcd507e59ca58d936ce218206ed03865201ce53ARG
NVIDIA_BUILD_REF
12/05/2023 10:36 PM UTC
611e1028f5af7681dc4fdf3b03f481eeebf962f25102b96a17b259509fac7b1aLABEL
com.nvidia.build.id=76438008
12/05/2023 10:36 PM UTC
689943238b479de8cdd89263d78b4964e06b766423bcd4ad705fe0834129add7ENV
NVIDIA_BUILD_ID=76438008
12/05/2023 10:36 PM UTC
4ff4683b3c7dcee37040f48a48985bed6782d11b637903bc98be106e46de5430ARG
NVIDIA_BUILD_ID
12/05/2023 10:36 PM UTC
8e00ae9a1242da326160ee8d2a7d58bd82240cd9b5753167a2a14d2efe912fc4COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
12/05/2023 10:36 PM UTC
ccf9ddc4361d1b703a35ca667de130444140660fe623bde9336ce5738c4c26bbRUN
PYVER=3.10 ln -sf ${_CUDA_COMPAT_PATH}/lib.real ${_CUDA_COMPAT_PATH}/lib &&
  echo ${_CUDA_COMPAT_PATH}/lib > /etc/ld.so.conf.d/00-cuda-compat.conf &&
  ldconfig &&
  rm -f ${_CUDA_COMPAT_PATH}/lib
12/05/2023 10:36 PM UTC
21e80174dd49b67f3324049cac784c44a8dac1c0171b1dfb18bc92bb3a8f59d4ENV
CUDA_MODULE_LOADING=LAZY
12/05/2023 10:36 PM UTC
5f247c9cf3ab09d3da6d5297dfe3c6d93ab521cbe441cd47210a12317e77a995ENV
TORCH_CUDNN_V8_API_ENABLED=1
12/05/2023 10:36 PM UTC
1e9ae6c0a295b4218a7c8d441f1d84c7ab3d663d9e17b283488b339024aeb474RUN
PYVER=3.10 if [ "${L4T}" = "1" ]; then echo "Not installing Transformer Engine in iGPU container until Version variable is set"; else pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}; fi
12/05/2023 10:36 PM UTC
...