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
1486b9ae01770b597a45260b29ff08ebc6748f4164e4026c9b0ef538bfa63641CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
04/17/2022 3:02 AM UTC
64c5b821dd55dc3a20684241d61b7c6a8982e5567462e7a72202e36d09a3cc96ARG
NVIDIA_BUILD_REF
04/17/2022 3:02 AM UTC
200f42aebb59e3ce61c4b37e9ee716c93515d2e9954e352c8c56e754a358b616LABEL
com.nvidia.build.id=36527063
04/17/2022 3:02 AM UTC
7a90f8880696d27925b390c15f6d799d7a18a4c91745b40bfd9d283efc98d6e9ENV
NVIDIA_BUILD_ID=36527063
04/17/2022 3:02 AM UTC
664421aa40ef5241b017629a4675e3f6dce81eccbd35a119e59ad7ab979cfb5cARG
NVIDIA_BUILD_ID
04/17/2022 3:02 AM UTC
b6ebfcbdcbc23ef6da137fe437d53084eb1446f13194b3a88b4c1dfe30a3e357COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
04/17/2022 3:02 AM UTC
7ae7cda3594aa1fb5fbdb36f850f0fa6571415d4d3124ce95b44b12355239c0bRUN
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
04/17/2022 3:02 AM UTC
952a9543885e39c5d50a682a7d99ab25c400150d86b2292d4924f614d700502eENV
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
04/17/2022 3:02 AM UTC
b6e536ee15488c6f65d56a8baf2dc27d10962df17e31ec8fe489b9a970090ec6ENV
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
04/17/2022 3:02 AM UTC
2b39f3266b0f0a5d42c9f4c9081031c98fdfb5a3dc11393c34d49cd2b959e981RUN
PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
04/17/2022 3:02 AM UTC
...