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
5a3d2e758296a81f9ca1d48dffde119ed74a5b4835de0a1d69f3357405fafd57CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
03/24/2026 3:43 AM UTC
1f4a0e9b6c46cc6ce2c30c52700bb22709fcc2c94e19846901713cb28bc36027ARG
NVIDIA_BUILD_REF=6a13c78105c69cd2cce4b1e6bfbed6ac5bfcee79
03/24/2026 3:43 AM UTC
3b2cbf0f31669c25a2d1703ad98599ccb7927e49f0ea377b8e04472bbe7b7735LABEL
com.nvidia.build.id=286725635
03/24/2026 3:43 AM UTC
e753a7cea43a95e4923567a8159d88a399830d3d88aa51855ab30e519804e2beENV
NVIDIA_BUILD_ID=286725635
03/24/2026 3:43 AM UTC
d245f6267c57524b328906a10f31479b9d0c4330feec019516c674ee53e5efd2ARG
NVIDIA_BUILD_ID=286725635
03/24/2026 3:43 AM UTC
15c1613206410c4cfa7b46aaf27c28707596395c51e6355f6ed8c65fec445da5RUN
NVIDIA_PYTORCH_VERSION=26.03 PYTORCH_BUILD_VERSION=2.11.0a0+a6c236b NVFUSER_BUILD_VERSION=67530f5 TORCHAO_BUILD_VERSION=+gitd9881220 TORCHTITAN_BUILD_VERSION=0.2.1+git71517cf6 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
03/24/2026 3:43 AM UTC
850a486cbe8cb3d41db45a3163a966c132d9b802596e60f4abb7e0fe9036b744ARG
TARGETARCH=amd64
03/24/2026 3:43 AM UTC
181c54d95baa4bc597b3c05d657289731edbd1e3ac51970b0e4dfc50d4963f35ARG
ENABLE_FIPS=0
03/24/2026 3:43 AM UTC
5e980e23e948ef8a646429460f78afe8b9b6846b12c49f09f5b87b6d33aaf0cbCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
03/24/2026 3:43 AM UTC
216e7a42ad94058511eb2454ffacf86fabfc4d5174106e40ad8b26e726ddb82eCOPY
constraint.txt /etc/pip/original_constraint.txt
03/24/2026 3:43 AM 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.