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
7e8c28faa7b998524623a9facc35a62baf90213ed314c508aa088e91d614c6abCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
10/12/2023 4:14 AM UTC
8e3ec599a2c4298476e11f0da4fc46c0749c548da8697dad5fd8f4d77acd131cARG
NVIDIA_BUILD_REF
10/12/2023 4:14 AM UTC
e115b0b2a28fd13da61e7523e509361e69ffe0ba7f34919ebc93e04f4c01e745LABEL
com.nvidia.build.id=71422337
10/12/2023 4:14 AM UTC
bf062a0814055ed85d876df4e157640aea5a9226a1da8dec355578118890a7d0ENV
NVIDIA_BUILD_ID=71422337
10/12/2023 4:14 AM UTC
9b43f31388ba0816063d2caaf23dff3cf03aaeb1825c34771d4bdba87663dab7ARG
NVIDIA_BUILD_ID
10/12/2023 4:14 AM UTC
89187794c8789f29222d0f834184855d8dc9d710a10d33a68c6ecb1c0d4a9534COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
10/12/2023 4:14 AM UTC
13b2fa1684bf0d0646dd7d8ac8cda2e4c92cc574f473181e15a76ca55573b981RUN
PYVER=3.10 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
10/12/2023 4:14 AM UTC
1e749f0cae1951c5e6f6631bdcf86aa6aa91e247d37f47adcacd60cb65329032ENV
CUDA_MODULE_LOADING=LAZY
10/12/2023 4:14 AM UTC
4235843f1d02f3eac8ea8ccc36ece70c3fac89257b748b00afcea3cbe5106f7cENV
TORCH_CUDNN_V8_API_ENABLED=1
10/12/2023 4:14 AM UTC
b85652ca017c26d2ab2ad55fffdaa40ac8ee75b4d2902cb2ae4e63bedada6c99RUN
PYVER=3.10 if [ "${L4T}" = "1" ]; then echo "Not installing Transformer Engine in iGPU container until Version variable is set"; else pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}; fi
10/12/2023 4:14 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.