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
324f4d2b5eb2fd77915384a9c651953ac61c22e6b75d48ddc8cade63a7fccfb5CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
06/07/2024 1:17 AM UTC
0b7d0935c92e984170f4f35ea3a7afc39dbb0c1642a7b6d6e8ad8cf5c3dbabf1ARG
NVIDIA_BUILD_REF
06/07/2024 1:17 AM UTC
d19668bd9239f42f1c4c35a62b5c7a1377faa39061d96fc860af91250925aed0LABEL
com.nvidia.build.id=96418707
06/07/2024 1:17 AM UTC
218100efcedd14bbc779391e45aafc4c23d75da8899ecd078adfda1991f696c3ENV
NVIDIA_BUILD_ID=96418707
06/07/2024 1:17 AM UTC
cc32851984629a41e5c322d1664f05115cb1e10c4dd829e771d03082d2272df4ARG
NVIDIA_BUILD_ID
06/07/2024 1:17 AM UTC
68317b9ca03c424467381971ea431d960a0bc47d2564fbc785e5a75643844726COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
06/07/2024 1:17 AM UTC
10a1d4952d25b5db021139b9abc9a6041d4f03107e21d5e279bd1b69909e42fbRUN
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
06/07/2024 1:17 AM UTC
1eadf755b73e94d584527661f1f142ed40687b7b757c0d5c03244dba3b6b92a4ENV
CUDA_MODULE_LOADING=LAZY
06/07/2024 1:17 AM UTC
6ac93cf341af3ec5e21421e905e6cee640040368abdec7199c862af3abf9f222ENV
TORCH_CUDNN_V8_API_ENABLED=1
06/07/2024 1:17 AM UTC
f937c7811816ce9711ea777272cb2baf16c5aa96dbe118d380e29fb8a851ec0fRUN
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
06/07/2024 1:17 AM UTC
...