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
f28d46ca093a82937a2c34d40362c6f134a1a98eabfa80c05f462ea0542e5692CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
09/14/2023 1:58 AM UTC
5dbf66d753c777acba5f425f54f18110a16ef51aa30d3d6c3c9aeab7a45f1a30ARG
NVIDIA_BUILD_REF
09/14/2023 1:58 AM UTC
8ed5a56b7d2e3299153994a8f761358536cf0eaf7bfd59679779cd7d2959e608LABEL
com.nvidia.build.id=69180607
09/14/2023 1:58 AM UTC
e3fbce14941ade739d6acb4a9e817e09325ad0f17540ec297ec57a89ce00db30ENV
NVIDIA_BUILD_ID=69180607
09/14/2023 1:58 AM UTC
f3dbe298386249c4f2b7fd4beb6a79c9cafb1684f71139315a5242e70fc0ad14ARG
NVIDIA_BUILD_ID
09/14/2023 1:58 AM UTC
6fc854809733187de2a59b5ba0e69860b916d6130a0b1dcd30debe4ee3bbadd3COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
09/14/2023 1:58 AM UTC
1e17006aba2de55629f58de550e33616890c604e6cae641c31bb619c16194212RUN
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
09/14/2023 1:58 AM UTC
d5b69c810476ea5b2754e8c88e110554c8a5c5f427eed2419288b589662f6939ENV
CUDA_MODULE_LOADING=LAZY
09/14/2023 1:58 AM UTC
9c10ba65afa1edb601843c2ad8b79d70c46a7bc381a775bae72aef3af534f77eENV
TORCH_CUDNN_V8_API_ENABLED=1
09/14/2023 1:58 AM UTC
ed37f507acf486441b158a464f3ec4006a4c93d881ea94d82af35685905a41edRUN
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
09/14/2023 1:58 AM UTC
...