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
6a12eba5aabf0910761088799dfb370e90c6fb815b647623695827df77064688CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
03/09/2024 4:30 AM UTC
f1236d89326a6488cf49c857adc984813a1bec379b704525f7ba4ce7e413cbb6ARG
NVIDIA_BUILD_REF
03/09/2024 4:30 AM UTC
f4ddcdc1281bed2db7a5d1dbaa79ab84b0cd57cf4609b0fe657b85cb1b6c4d75LABEL
com.nvidia.build.id=85286408
03/09/2024 4:30 AM UTC
de58882a13e184c49d3746445aceb2ca16a9aa8082116cc26634f92b80764cf3ENV
NVIDIA_BUILD_ID=85286408
03/09/2024 4:30 AM UTC
2f09b9c4f5ee45331d2ca3c5e0f62f72ed75ac52bad5dc53fc6031b2f5db6175ARG
NVIDIA_BUILD_ID
03/09/2024 4:30 AM UTC
704fc2b0f6d43d830969a4cf26598a25a9669690317f323abea64a86404d7f4aCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
03/09/2024 4:30 AM UTC
66109029527765ec01fcf871a2da80d01422d9856250d7f1788c3a3347cc1bf9RUN
PYVER=3.10 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
03/09/2024 4:30 AM UTC
b78f4eca08167c00c599025ef18b33ffe4b2afabd420039f191343431fab2c50ENV
CUDA_MODULE_LOADING=LAZY
03/09/2024 4:30 AM UTC
9977f1d7d9148ad7c6582b3f3027351bb9c28192bf000daaa2593e43e1859385ENV
NCCL_WORK_FIFO_DEPTH=4194304
03/09/2024 4:30 AM UTC
2476bcb053a359fb95d2a7e5db907f0457a94ae318414984c03ed0e56f2b7f5fENV
TORCH_CUDNN_V8_API_ENABLED=1
03/09/2024 4:30 AM UTC
...