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
7e143625382967f2000c5b776b082ef98ba93c27a9844c61ae300703caf9fe95CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
06/14/2022 4:43 AM UTC
8b5883f6f20fa2400184fc788687ebb05180d0be4c0d728a876fea00d64bd8e4ARG
NVIDIA_BUILD_REF
06/14/2022 4:43 AM UTC
f67135e4ec39b2ebc4111188d61edaadd5a8600ed50932919f6fca6006bb5b4fLABEL
com.nvidia.build.id=39176199
06/14/2022 4:43 AM UTC
1a576aaa8f8f395315741b010d45f97b083c6d156cd2ba4900840e001ab30a92ENV
NVIDIA_BUILD_ID=39176199
06/14/2022 4:43 AM UTC
365f8a2692293b49566de724de30f702ee4af6f5fccf1ea81f4408600f56edb3ARG
NVIDIA_BUILD_ID
06/14/2022 4:43 AM UTC
67698cbd8a8278a5ca1b5a5f30b732d4ba440adf4f4ddb9dfeb1d7c9abbead79COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
06/14/2022 4:43 AM UTC
4b22f7c2ac052b736c1c7dd91c955c902e355d847bad9c0e8996eac0febe0554RUN
PYVER=3.8 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/14/2022 4:43 AM UTC
daa93df6f8dd8c6df3667fe93668eb7a0b102d4f5f9b5825cb4c3c6d0496c5caENV
PATH=/opt/conda/lib/python3.8/site-packages/torch_tensorrt/bin:/opt/conda/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin
06/14/2022 4:43 AM UTC
2020ebb6c458ed6b6b0334bcbad44dcf82f6eedbbf90b5fcb1057d3ecbc47ed4ENV
LD_LIBRARY_PATH=/opt/conda/lib/python3.8/site-packages/torch/lib:/opt/conda/lib/python3.8/site-packages/torch_tensorrt/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
06/14/2022 4:43 AM UTC
b2ce44e1f06f68132dafc591503214cf5f16e4264d57a50350a925df8eff3eacRUN
PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
06/14/2022 4:43 AM UTC
...