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
f0a1e76f7af35aeb5075a9b2b1e8917e08d9ceaa78c5d08ff5a6c4e825f1a8ffCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
11/20/2022 6:30 AM UTC
160be1a92aeba89b249f1cc440b50ebee37ae5b47527a37bb9fb8a99774b3440ARG
NVIDIA_BUILD_REF
11/20/2022 6:30 AM UTC
9c2e6b97265973be1f91ee824d0c6ee9955dc8c6f99d23a50482067f1a2adecbLABEL
com.nvidia.build.id=48503342
11/20/2022 6:30 AM UTC
5a03fd18f3e25b27012840e75f275668fa2704c6582304e0369e1a06555e16abENV
NVIDIA_BUILD_ID=48503342
11/20/2022 6:30 AM UTC
4c9ff32af7fb93b470115a9ad9665e697933f361bda75727f30716fbd0144babARG
NVIDIA_BUILD_ID
11/20/2022 6:30 AM UTC
e701dfb038aed50c22fdcc0fb4bf437c41ee1ea0be6e3532616fa0660e525f54COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
11/20/2022 6:30 AM UTC
bac0dfb810b9c6b7018db85b2d94fb31dd8ab5a09a1f770e0097d5cc756fba5cRUN
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/20/2022 6:30 AM UTC
07e3745f6d9e47b1c2d5f89f32562397dcd3a5c644286d0b627e8bdb145eee77ENV
CUDA_MODULE_LOADING=LAZY
11/20/2022 6:30 AM UTC
27d3b92e32d08df7415af8d9b6ec132d0585d49e53bea15e7e45409d1e68201fENV
TORCH_CUDNN_V8_API_ENABLED=1
11/20/2022 6:30 AM UTC
295a2fd4a2905a6ec73fa49b92a9db795b997bc1e361dddc5d05352988777071RUN
PYVER=3.8 pip install --upgrade git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}
11/20/2022 6:30 AM UTC
...