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
a2fa59841b1c82876365994ecaedf203fe150118b6c23a22ea6136318f051b64CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
11/15/2024 6:54 PM UTC
e5fe60b5d92d22edfb7cf7a2a54fd10528e4026633e3f45c576573d91f6b67abARG
NVIDIA_BUILD_REF=40d6cf911ff7816d89ec212ae4ba065395be1d07
11/15/2024 6:54 PM UTC
1aca583b0379fdbbaf9d661087b57b27f0fa60691b470124e6c8f5d12cf7617fLABEL
com.nvidia.build.id=122402533
11/15/2024 6:54 PM UTC
7fb13600de229c6786be6703fd7eed44f7cce03306c0cf5e8c0ebf8dfd7059f8ENV
NVIDIA_BUILD_ID=122402533
11/15/2024 6:54 PM UTC
5f47b1f163d1bf0aa788e33c86b2341ea061fe37d0c1a21a9624d9958bf3abd3ARG
NVIDIA_BUILD_ID=122402533
11/15/2024 6:54 PM UTC
27b8227437fffd6c1fc7f65178a41861bc7a4cd97e78bbc7875714a4c52565ccCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
11/15/2024 6:54 PM UTC
e711405558fbc6edf82c38180d9cebd67395ea48f13065985b2eaf5f89beb862RUN
NVIDIA_PYTORCH_VERSION=24.11 PYTORCH_BUILD_VERSION=2.6.0a0+df5bbc0 NVFUSER_BUILD_VERSION=0d33366 TARGETARCH=amd64 PYVER=3.12 PYVER_MAJMIN=312 L4T=0 PYVER=3.12 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
11/15/2024 6:54 PM UTC
24bdc083cd276c05223a364f62746871d2b5b03e9c7a20850319a79ab476549cENV
CUDA_MODULE_LOADING=LAZY
11/15/2024 6:54 PM UTC
b9ffa98d0cf2ccde1a7529a19de8fadcbf132f249526944a7d147bf26d0085e1ENV
TORCH_CUDNN_V8_API_ENABLED=1
11/15/2024 6:54 PM UTC
3d330575fad0aea4133fcf288e756b3ca77801102c608982e557f042fdf2fb3dRUN
NVIDIA_PYTORCH_VERSION=24.11 PYTORCH_BUILD_VERSION=2.6.0a0+df5bbc0 NVFUSER_BUILD_VERSION=0d33366 TARGETARCH=amd64 PYVER=3.12 PYVER_MAJMIN=312 L4T=0 PYVER=3.12 if [ "${L4T}" = "1" ]; then echo "Not installing Transformer Engine in iGPU container until Version variable is set"; else NVTE_BUILD_THREADS_PER_JOB=8 pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}; fi
11/15/2024 6:54 PM UTC
...