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
3e6cd82e1199a1c68b445570a7931645b7a1616d544e09ed956a1cc42af106a3CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
08/08/2025 6:31 PM UTC
23a4f3a9eb46e7d8084e2e9a3a29b4d4ccb3857b5a2d763a4028b341cc8164dfARG
NVIDIA_BUILD_REF=f7207771a583a8091f425dfdbcaae60bbce7fe76
08/08/2025 6:31 PM UTC
9e52312a4e815b492a664d047f2b8b1fa7d71b07261733810ee9a7a82718283dLABEL
com.nvidia.build.id=197421314
08/08/2025 6:31 PM UTC
64ab07d5d47a7b70b56edd8465a0420f09adf76d1d6ba6cf57175caf13e4d595ENV
NVIDIA_BUILD_ID=197421314
08/08/2025 6:31 PM UTC
eed18e18038b7042ff622db17c50576e89ca1645d8d01f97769fbd0642012e03ARG
NVIDIA_BUILD_ID=197421314
08/08/2025 6:31 PM UTC
7131297291e2a74c047a2541c5c47820c8407227fd40ee93952f0a7a78a7b3a0RUN
NVIDIA_PYTORCH_VERSION=25.08 PYTORCH_BUILD_VERSION=2.8.0a0+34c6371 NVFUSER_BUILD_VERSION=1a27c18 TARGETARCH=amd64 PYVER=3.12 L4T=0 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali cd /opt/pytorch/pytorch/ &&
  sed -i "s#    if (props.computeMode != cudaComputeModeDefault) {#    int computeMode = -1;\n    TP_CUDA_CHECK(cudaDeviceGetAttribute(\&computeMode, cudaDevAttrComputeMode, device.index));\n    if (computeMode != cudaComputeModeDefault) {#g" ./third_party/tensorpipe/tensorpipe/channel/cuda_ipc/context_impl.cc &&
  cd -
08/08/2025 6:31 PM UTC
e6d8ea25fd7eb109a6ae1efc87d5d7ee16021f76b00669f231ea486670c8dca9COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
08/08/2025 6:31 PM UTC
4794fec428a39bdcef5205fa42137ac4bd25449c546e0984a6b16d47310fd6a2COPY
constraint.txt /etc/pip/original_constraint.txt
08/08/2025 6:31 PM UTC
93806be98a197b59b2467cf5cd06192c1d01778fdb722591d728de09d58a16bfCOPY
restricted_constraint.txt /etc/pip/constraint.txt
08/08/2025 6:31 PM UTC
a759105194b3c9ad83b589b9e9f058f4edbe78410de05541a94fafef95d196feRUN
NVIDIA_PYTORCH_VERSION=25.08 PYTORCH_BUILD_VERSION=2.8.0a0+34c6371 NVFUSER_BUILD_VERSION=1a27c18 TARGETARCH=amd64 PYVER=3.12 L4T=0 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali 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/08/2025 6:31 PM UTC
...