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
855b7f8da28e5c26cb4f8937c652ce0fc97e7e8279673e5e435181a29e3bbf23CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
08/19/2024 7:02 AM UTC
b137a8c7c5ddda6217d19f0a22dba4304a3577a4143b680034895eb15b23dd4cARG
NVIDIA_BUILD_REF=25aae698093a80672c33aac448393478894f210e
08/19/2024 7:02 AM UTC
7c4b5c9d9401dcef275d68dffe0e00867cdced67485bc044aed0c0adb2bf20b2LABEL
com.nvidia.build.id=107063150
08/19/2024 7:02 AM UTC
2e28b5d2a835ee3824c26216e741f80c4e81673fa76caaa547ac3a5b3d1cacb9ENV
NVIDIA_BUILD_ID=107063150
08/19/2024 7:02 AM UTC
516359eeb829936264b065c86938a1984137878e401690a6d716dc50d1b66e64ARG
NVIDIA_BUILD_ID=107063150
08/19/2024 7:02 AM UTC
af5ae5d9c27226eb5b53afe1bd92deb4b105b1a4a22da3840bbde3a11d5a8fb2COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
08/19/2024 7:02 AM UTC
64f89d3ba8ef0ed41edb55eea5ef709e6eafc29f631b35a1c06c72c921c91d76RUN
NVIDIA_PYTORCH_VERSION=24.08 PYTORCH_BUILD_VERSION=2.5.0a0+872d972 NVFUSER_BUILD_VERSION=1d02b13 TARGETARCH=amd64 PYVER=3.10 L4T=0 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
08/19/2024 7:02 AM UTC
1e803c027b3e409dc549a2f5a11ed4b17a659278b6b97306e1eb38c5b535c36dENV
CUDA_MODULE_LOADING=LAZY
08/19/2024 7:02 AM UTC
ed2c3cdffe7d79e49db40f0b86df7df3d1aa4a7bd8e9865d08f8694f788cca82ENV
TORCH_CUDNN_V8_API_ENABLED=1
08/19/2024 7:02 AM UTC
367cab4bbade24dc15d2f78dd3f12a3ffdfc5636a423aa759d0eda8b709030deRUN
NVIDIA_PYTORCH_VERSION=24.08 PYTORCH_BUILD_VERSION=2.5.0a0+872d972 NVFUSER_BUILD_VERSION=1d02b13 TARGETARCH=amd64 PYVER=3.10 L4T=0 PYVER=3.10 if [ "${L4T}" = "1" ]; then echo "Not installing Transformer Engine in iGPU container because 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
08/19/2024 7:02 AM UTC
...