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
b46d92ef8238178fd27882b921a822cd2afa002c2310eff62e83f747504d4547CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
02/20/2025 1:38 AM UTC
68982a1de28794f8d1c0cba8c401c28c54bc0f64a019cffd97b5ab58167bee0dARG
NVIDIA_BUILD_REF=2235e97cfdf0c73e377b69c5420a10e4efa84fe7
02/20/2025 1:38 AM UTC
b22fa36e1ef59c03b97fe2e1308849201d19fcf5c272e1b63616b2fe3fa6c445LABEL
com.nvidia.build.id=143088496
02/20/2025 1:38 AM UTC
c909f2cf49d7d0f7ec7e9990d88185f86636487b5a9ba0dd369057f1aa8098c8ENV
NVIDIA_BUILD_ID=143088496
02/20/2025 1:38 AM UTC
b3e4a2f31850db3fb1e1f151977724508066a797383efea8ea44bf52ffba4c63ARG
NVIDIA_BUILD_ID=143088496
02/20/2025 1:38 AM UTC
2d9c94d80cee2755f60fb36732fe98744d0c578fae905c6dc3331b954557a9b8RUN
NVIDIA_PYTORCH_VERSION=25.02 PYTORCH_BUILD_VERSION=2.7.0a0+ecf3bae NVFUSER_BUILD_VERSION=6627725 TARGETARCH=amd64 PYVER=3.12 PYVER_MAJMIN=312 L4T=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali cd /opt/pytorch/pytorch/third_party/gloo &&
  patch -p1 < /gloo.patch &&
  cd - &&
  rm /gloo.patch
02/20/2025 1:38 AM UTC
25f811161577aee5471b5c4c7e82113c5265abbae62c262694b874294d1b3217COPY
gloo.patch /gloo.patch
02/20/2025 1:38 AM UTC
d46a98343d104ef049b7f5de6b2fcf679119cbe49a22ee1bdd8bb1f9936be19fCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
02/20/2025 1:38 AM UTC
90462c6300eb200442b6b65aa6d4752e5927b4a4d392b65bd1f1c145e1aeafdfRUN
NVIDIA_PYTORCH_VERSION=25.02 PYTORCH_BUILD_VERSION=2.7.0a0+ecf3bae NVFUSER_BUILD_VERSION=6627725 TARGETARCH=amd64 PYVER=3.12 PYVER_MAJMIN=312 L4T=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
02/20/2025 1:38 AM UTC
c87b6169599659c1cde89d8f43ee0e249e6dd946ecbe29c5d886cf04f3c2f4caENV
TORCH_NCCL_USE_COMM_NONBLOCKING=0
02/20/2025 1:38 AM UTC
...