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
125da16cbb611b9cdeec942a0d0246ceae3ab76587bfe28443d5741ae97214f5CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
01/25/2022 11:05 PM UTC
7ee6c8e30eae02f45bb20ca2813dd172b2b5701fa80927e9c170f34692d191ebARG
NVIDIA_BUILD_REF
01/25/2022 11:05 PM UTC
739e130f8650c3af7bec2e06810f3959f6465199cef3ca89624ba46757800742LABEL
com.nvidia.build.id=31424411
01/25/2022 11:05 PM UTC
c4b294943c3a7e8812a70902a0b068391eca20c43e7deb048af76fdea540a391ENV
NVIDIA_BUILD_ID=31424411
01/25/2022 11:05 PM UTC
4cec0a4c433d58d7ba029e5deed6699b4c5abe7495c570e1d5f470f12d067cb1ARG
NVIDIA_BUILD_ID
01/25/2022 11:05 PM UTC
933ed3f4a3c488b11ff5f6facbe2e30df0e4709beaffb970995be03cf37538ddCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
01/25/2022 11:05 PM UTC
fcb036804cda7a8ab73f71409ab2711834d9813140f3005362600e4966d321a1RUN
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
01/25/2022 11:05 PM UTC
79f9c592cb733271a0667f5a6a6f78dfac2f480ad0a5c0febf713ada461d1ce5ENV
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
01/25/2022 11:00 PM UTC
6c58437996baab366ae075d2aebfec02ef72e7ac81ab39a7eccb051f255c9063ENV
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
01/25/2022 11:00 PM UTC
8738b01a3b4c90412b98a4dfe278e2b0681fc7ff6019d8c171b58b0b856014b5RUN
PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
01/25/2022 11:00 PM 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.