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
342c2f6190d951ffc1e3893e0d224bb6462fa1e43b1772152e4a9f8f7cf9a48bCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
11/12/2021 1:21 PM UTC
63e1fb9d617001737716740e82c853d1a65aea512f97133659ead7640e61f820ARG
NVIDIA_BUILD_REF
11/12/2021 1:21 PM UTC
7122ab970f5e19c5b70b33440c7f3b21b2276d3b2e444b9ca9bd220a67c91f6dLABEL
com.nvidia.build.id=29224839
11/12/2021 1:21 PM UTC
0e8c869e8506c08102f3eaec33c3831cc897ff1eded5d2098fb8457538e02603ENV
NVIDIA_BUILD_ID=29224839
11/12/2021 1:21 PM UTC
3704daa6dc17e212de61d5dbd70846a755e5e2ec1ffcf824db0348f47c4f211eARG
NVIDIA_BUILD_ID
11/12/2021 1:21 PM UTC
08d2ef5d18ec64bc2488f669aa242401cb6040a4a902433b789639931c4586d7COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
11/12/2021 1:21 PM UTC
2d49a35049d277bc47e0d49b73f01c7e558f9a55925df9caca1ca30c1a49bf00RUN
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
11/12/2021 1:21 PM UTC
f04094d818cd33bf21b3d6da84a17542c04d54c1e7f78634465f3bce15d202f7ENV
PATH=/opt/conda/lib/python3.8/site-packages/torch_tensorrt/bin:/opt/conda/bin:/opt/cmake-3.19.3-Linux-$(uname -m)/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
11/12/2021 1:16 PM UTC
a5295c8430ca31fa9244ee3ac6545bd079f334ac61ae9854bd1d7ce297a87008ENV
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
11/12/2021 1:16 PM UTC
5fa5f85ca23d856585826732cb1403b8da6b447eaec24df7f0a254ffec4c9032RUN
PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
11/12/2021 1:16 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.