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
6e52ac2ed91e2f4e19de7d3de19d95cf9c304dfe2186b8032f73539e265e78e8CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
12/02/2021 8:07 AM UTC
643a56531a0da317e37be5f448506edd6b1310f66b2e73721893559028f32eebARG
NVIDIA_BUILD_REF
12/02/2021 8:07 AM UTC
ae366ea1801eed31507e5e8e74cc553371bdc1a7f68412e264b4c20fd2f9a824LABEL
com.nvidia.build.id=29870972
12/02/2021 8:07 AM UTC
73dff87f12067cb903dd48a41d14cf884d226b405ab81bd16a581b139d6aea01ENV
NVIDIA_BUILD_ID=29870972
12/02/2021 8:07 AM UTC
61d4d43c760519cab5fb8d2928ef6adb0b061b48f5f2ab1506715b2d75d1a722ARG
NVIDIA_BUILD_ID
12/02/2021 8:07 AM UTC
cd6daa77f1c554f24abb6f82158508fe8b4d09d0bc30e96185a0543f9a9fec6cCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
12/02/2021 8:07 AM UTC
549ca66433295bd0ee73d1273bf323f2e8cd2e891b245f8280838ffdddd7b8f6RUN
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
12/02/2021 8:07 AM UTC
d82feb4bb7d601368f00ce830af4e74a5e13dce12259b00551674078c9cde446ENV
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
12/02/2021 8:02 AM UTC
5cad7151382c5922120179b31bf5df7742100bc832df35e19459774c87c3a846ENV
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
12/02/2021 8:02 AM UTC
580ff3013797402be1e4d4b1d03ddaf50a067b83144c6d4b44a928027991b5f1RUN
PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
12/02/2021 8:02 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.