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
44313eeeb7918ec4f6e6a4525fe1470811d556fa8b76d46ef1e2cc27f2e34d12CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
08/04/2022 7:45 AM UTC
2fdd7e3d67d80fde92e063dcc774a26b71546272a3682e5e1d20500a80b0db18ARG
NVIDIA_BUILD_REF
08/04/2022 7:45 AM UTC
da806c33806b39e6b87caafb5bf38ea4ede45a675dd2daeec2842f5a374ce5dbLABEL
com.nvidia.build.id=42105213
08/04/2022 7:45 AM UTC
ced2ec984e1a0487b52adc5fe0d5309d72391237f55d17eeb15130695042694eENV
NVIDIA_BUILD_ID=42105213
08/04/2022 7:45 AM UTC
bc8efe8b104846a7b1dae5ffd0b123df5be503a943b49353563a2fabe97af215ARG
NVIDIA_BUILD_ID
08/04/2022 7:45 AM UTC
b87e970997000db7f306908d294c7ca53089fe5dd5683d119262212a87bcb250COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
08/04/2022 7:45 AM UTC
e3f2437e24d92157cc323f7412a90c6c8c6584c21bcb5885aba3ef23a6ec3c12RUN
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
08/04/2022 7:45 AM UTC
31e4ce860fabb3b36ed2a4fbdbe447ec8c37966f2393f472df13bd65fd8a3625ENV
CUDA_MODULE_LOADING=LAZY
08/04/2022 7:45 AM UTC
733d9deb8654ca4694f714c76f4d33835704613bbc3c68c47b36a6bf103b8dceENV
TORCH_CUDNN_V8_API_ENABLED=0
08/04/2022 7:45 AM UTC
edb2bd66f8f89d0466a014de61cde572e8ab3ea28a45bb4b84d8e6fc484e3a6fENV
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
08/04/2022 7:45 AM UTC
...