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
eca2973e1c9b03de489e8cfe060d624b8db0199eae60cb11e929bdba4ae7cf54CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
02/10/2024 2:48 AM UTC
14e25daad7ff1e721a48fa75374b22ac8b023ea34c439a2019776e877a486c92ARG
NVIDIA_BUILD_REF
02/10/2024 2:48 AM UTC
dd1bfabce220e2992f2f69d8ce31df73d5f9c8bfa63c50b912795b93db4b6b70LABEL
com.nvidia.build.id=82611821
02/10/2024 2:48 AM UTC
7a55858027e4bbb80bb58479982bccc1560a08894bf1f3efaa3fded5379b6b76ENV
NVIDIA_BUILD_ID=82611821
02/10/2024 2:48 AM UTC
e7e60242974e0cd8f52fadb4025b8f8c8255c7593a5e788af10c55d9c3854018ARG
NVIDIA_BUILD_ID
02/10/2024 2:48 AM UTC
bf00b3b037836ad9c752343abfd0f3aee7b15d5821d33c780779894e789c455bCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
02/10/2024 2:48 AM UTC
9ee453b1a5b234bb49dfa4b79a1fcb04da67488a61e7a51520e51815ea28d655RUN
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
02/10/2024 2:48 AM UTC
b282472809021d8c755b168dce210f175aa9c15c12ff481d3be011b86cfdeee8ENV
CUDA_MODULE_LOADING=LAZY
02/10/2024 2:48 AM UTC
9081a2c1488f9e7e1e8b864164c79a13e4bfe0c3a3bf6218e95ef434d6c2ee23ENV
TORCH_CUDNN_V8_API_ENABLED=1
02/10/2024 2:48 AM UTC
7f4cd2a2e297805724160bec1e66d781deb5072e50da3c655db2fcaca4d0c944RUN
PYVER=3.10 if [ "${L4T}" = "1" ]; then echo "Not installing Transformer Engine in iGPU container until Version variable is set"; else pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}; fi
02/10/2024 2:48 AM UTC
...