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
9345f730b0b5759437a2f7a27bd28c92a7862f4613b378a019b65ab88e4ba87fCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
03/22/2022 7:02 PM UTC
e0897b1d4ab6e4768a0a673dd56e3e09193c975afca9168c410ab1505b8aef9cARG
NVIDIA_BUILD_REF
03/22/2022 7:02 PM UTC
5ff1b2595f4d9d6867360c3c56df6dd6af139c10647dfa52ec59f7c8f5c2f5f1LABEL
com.nvidia.build.id=33569136
03/22/2022 7:02 PM UTC
7059761e2fb406477fbb992e9c22f6a0c6ab65a46bf1d4dc78e222e4a0f96085ENV
NVIDIA_BUILD_ID=33569136
03/22/2022 7:02 PM UTC
b643fd20352a5e25e31ed8a8dc7a5f37dc65a9723725632db0e788a0d09e3110ARG
NVIDIA_BUILD_ID
03/22/2022 7:02 PM UTC
1cefb66f1eb1fb490ef1c5d71ee65ec5ddc1efd46c11e982cc671c476f315bb5COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
03/22/2022 7:02 PM UTC
fcce521df666e8300d48c449139af70311d78b4a974cfc9bb068be67050cf53dRUN
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
03/22/2022 7:02 PM UTC
1cef113d5151bb4f0e925300fb1f47b9fb974e9a8a520bf454a9ae1cc0a1bb5eENV
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
03/22/2022 7:02 PM UTC
59f355cfefd85f4dd4b3fd07d85ff39b6b6c8ccce409fb8af5e4df532e78aeddENV
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
03/22/2022 7:02 PM UTC
2aeeabc068c90e88110f305e487ea3005cb11665b6bbfd4bf1866cdeaaadb4c7RUN
PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
03/22/2022 7:02 PM UTC
...