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
7552184dc0b9b2921176a76c426e6cff43b9d1c442367272de2a9d561ab045afCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
07/05/2022 10:20 AM UTC
56517e289be7d4663e8cd13c77ac0bb2f664ec353862e76c312a174dd5ef4aabARG
NVIDIA_BUILD_REF
07/05/2022 10:20 AM UTC
1e02e8246b550ccb2446b6abc420d18d3882d90d3a9e4a26ee9b03c5c8ec4bfbLABEL
com.nvidia.build.id=40241807
07/05/2022 10:20 AM UTC
0b2d5654a994e4837c76fb332c3be0eed80d718cefe44a1778486ff09d07208bENV
NVIDIA_BUILD_ID=40241807
07/05/2022 10:20 AM UTC
1c76abefdd52323283f0197ac3e694d9943df3ceb72967c7a6ceb2950b159de7ARG
NVIDIA_BUILD_ID
07/05/2022 10:20 AM UTC
8078beda4edc90ae6bfa424a93ee12139fae0d16a2ac5496ea908397b7a30bc1COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
07/05/2022 10:20 AM UTC
577f15771d44490cdcbd88d77492880f9de4b9b2bda33a1da4ee10d6584dcea4RUN
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
07/05/2022 10:20 AM UTC
a9d4c11c63c9f6c0df16dae0238c86192ab9138a833ceaa28e3698a71c144d83ENV
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
07/05/2022 10:20 AM UTC
99ef46c2ef39be966e45658e68a576e063fdb5a33f5bfed7e745705e09a4367fENV
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
07/05/2022 10:20 AM UTC
5c4a60eb833b88093fdf8d52e96d9ef0b5fed68dcdb55d3a334ef2dc723cbc21RUN
PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
07/05/2022 10:20 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.