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
4b4fb9d9c2cc4088b41455a8d957597cc6e5c4527d39d6b819af07510b5dab3aCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
09/17/2025 7:07 PM UTC
ca8e8fb7fc450052e09cf7ebc90fcc255bb66be625967e7e5be5caeb69216135ARG
NVIDIA_BUILD_REF=0abbc99c3a779a3f40a8b337e832a3a66d3f1e6e
09/17/2025 7:07 PM UTC
4d1ab2ce71d305bf48b7d94940acd6383294ed7cbc0e2f51a72de5deb9adfb5cLABEL
com.nvidia.build.id=210907837
09/17/2025 7:07 PM UTC
e4a24ae524bc2cc2a252f18611ade8ae54e3018e8c486089738263b1ad45b990ENV
NVIDIA_BUILD_ID=210907837
09/17/2025 7:07 PM UTC
6c3e68106eef976fcc7dac4a5f801d0c0f04beb816c1a75ed09fbd0f59f79983ARG
NVIDIA_BUILD_ID=210907837
09/17/2025 7:07 PM UTC
163d589375a08a2d5e882a537593f825215a6c17db521918bdb8defe0711e79eRUN
NVIDIA_PYTORCH_VERSION=25.09 PYTORCH_BUILD_VERSION=2.9.0a0+50eac81 NVFUSER_BUILD_VERSION=a71c674 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 -
09/17/2025 7:07 PM UTC
27021a6a5403a467c8352eb17099aa4aa6a952dc56f9c33a0f67fe1cb426f588COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
09/17/2025 7:07 PM UTC
fe3bb1baad829f72145602971e4ebb0ea458c8105ccecca2a3f0b6f4218c127fCOPY
constraint.txt /etc/pip/original_constraint.txt
09/17/2025 7:07 PM UTC
231eef8c09e073aad33d9f466c7eb18bbfa30b9de27506fb5c3cf7fe67661c77COPY
restricted_constraint.txt /etc/pip/constraint.txt
09/17/2025 7:07 PM UTC
8de7d9c2f162781da63bf177a08612c9f0b74e9bcff57f9e848377758ebb454dRUN
NVIDIA_PYTORCH_VERSION=25.09 PYTORCH_BUILD_VERSION=2.9.0a0+50eac81 NVFUSER_BUILD_VERSION=a71c674 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
09/17/2025 7:07 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.