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
766d974622fb30cf84c7e413af76cf17d0e2c700f2b2cf53a99cda63fe0db86cCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
06/09/2026 11:57 PM UTC
20af6ebcd8427e85a1aaa2efa5d3cebbbc47b43795cd118a843c947ee6d6b86fARG
NVIDIA_BUILD_REF=d557151f4c7ddca284cb5e8d5ce78cee4d80f7e5
06/09/2026 11:57 PM UTC
29996e7c299a64662cd3b6c357bb2d6781ba0de9ea2757e942bc514f8c5d9a3eLABEL
com.nvidia.build.id=337426143
06/09/2026 11:57 PM UTC
88d678a91ea8c8970a114017159729e30d8ed7a22c912eeabbd893218e18116eENV
NVIDIA_BUILD_ID=337426143
06/09/2026 11:57 PM UTC
1249daf5c87f53742e63980a51d16401156b0d7a69721c5eef894d21f3d23e64ARG
NVIDIA_BUILD_ID=337426143
06/09/2026 11:57 PM UTC
f92dd03bc93d5f1d554e5919d5c1f99b94297636804929bdba8ccb50198af457RUN
NVIDIA_PYTORCH_VERSION=26.06 PYTORCH_BUILD_VERSION=2.13.0a0+8145d63 NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git28e6aca5 TORCHTITAN_BUILD_VERSION=0.2.2+git2ae13405 TARGETARCH=amd64 PYVER=3.12 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali ENABLE_FIPS=0 if [ "${ENABLE_FIPS}" = "1" ] &&
  [ "${TARGETARCH}" = "amd64" ]; then echo "Running fips-fix.sh"; /tmp/fips-fix.sh; fi
06/09/2026 11:57 PM UTC
ea6317a3f85978e66aec804f65f6714a6786ec2d7832bb818dcbc7c1474abfbfARG
ENABLE_FIPS=0
06/09/2026 11:57 PM UTC
db0752d30ac07b1ce585f49c246f964191b82178531512f0280ba22d67f3e085COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
06/09/2026 11:57 PM UTC
901486900986ad3ebb2ec7effe9d766f7311de07290164a05168e6893f4cbbd1RUN
NVIDIA_PYTORCH_VERSION=26.06 PYTORCH_BUILD_VERSION=2.13.0a0+8145d63 NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git28e6aca5 TORCHTITAN_BUILD_VERSION=0.2.2+git2ae13405 TARGETARCH=amd64 PYVER=3.12 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali cp /tmp/restricted_constraint.txt /etc/pip/constraint.txt &&
  cp /tmp/original_constraint.txt /etc/pip/original_constraint.txt
06/09/2026 11:57 PM UTC
096d848c9fabfb57f9bd2e7cfeafbcf300cc103f0cc26a9649175df39faea3eeRUN
NVIDIA_PYTORCH_VERSION=26.06 PYTORCH_BUILD_VERSION=2.13.0a0+8145d63 NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git28e6aca5 TORCHTITAN_BUILD_VERSION=0.2.2+git2ae13405 TARGETARCH=amd64 PYVER=3.12 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
06/09/2026 11:57 PM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.