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
9932b4854e96fe7bdda8918f59bf431e2a29a9a5e9b2ba7f840300a373d909d3CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
02/15/2022 9:39 AM UTC
2d022b1c4e609592db6a821540262377303dec54d274f37512a7d87e1c59a533ARG
NVIDIA_BUILD_REF
02/15/2022 9:39 AM UTC
d348b587d1757de7e7486d2566a86d38ce293146d021b55b598ac8a1e979c7d0LABEL
com.nvidia.build.id=32255746
02/15/2022 9:39 AM UTC
f348d096a2c91cb97745237a515db424d2c5641348e3f637b6b0433c8b10c621ENV
NVIDIA_BUILD_ID=32255746
02/15/2022 9:39 AM UTC
a768615048ba393119429bcd98cdaa2df82d7f4c55b8538fc4990171efd40015ARG
NVIDIA_BUILD_ID
02/15/2022 9:39 AM UTC
b4266a29aa8067c7da3f8b0485e5362ce3df01b91afb6b4a51232283502799f0COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
02/15/2022 9:39 AM UTC
8562f231f90ae75505fa7d5f1ccc2cd172b2ae3cc0daa6e43c8c76f4d17b88f8RUN
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
02/15/2022 9:39 AM UTC
bacc3ba0c295f858c61ecf411e9af8918cd9a42f5b29904a39180dafdad771a0ENV
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
02/15/2022 9:33 AM UTC
f62db0b4e16859fd838adb3d631304c2cd1a90761bc8f8d1016df2b992475dccENV
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
02/15/2022 9:33 AM UTC
053b951e00d79e6a903603826b997e37a9e019db431541a4297b89c0f306600dRUN
PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
02/15/2022 9:33 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.