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
cef878e331c27c5ff0da0985187814d2fd551d47abcd828cda000760d7c86c9dCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
01/25/2024 5:13 AM UTC
a5c8e809791f61996cdceddf21d635b94aa86b8411ecd9ff49acde23a17029b8ARG
NVIDIA_BUILD_REF
01/25/2024 5:13 AM UTC
4d06053296da76289d7040c94cd4ea7ede712db2b1f446303a6ee3bcc98e063aLABEL
com.nvidia.build.id=80741402
01/25/2024 5:13 AM UTC
3b770b98765e845c240dcb4fa62383122ae6b3171d833266df8fd90138c8db19ENV
NVIDIA_BUILD_ID=80741402
01/25/2024 5:13 AM UTC
a52caa32e7227083d035aa800a3ae41ccf925173ed34f3b1eb03eef0f2a91d65ARG
NVIDIA_BUILD_ID
01/25/2024 5:13 AM UTC
0082c58dad6ea2436d746a5afea4457e611bda232e15b444f3bf18df631077a3COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
01/25/2024 5:13 AM UTC
d83ec90c4f96cbe76ebc3c084f072259339b8e8791fc1207ebccc3bc361332f9RUN
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
01/25/2024 5:13 AM UTC
3a2d1f741db94ef982d4484542ffc1d80ddb0fc3009c91097710d7346c8f1888ENV
CUDA_MODULE_LOADING=LAZY
01/25/2024 5:13 AM UTC
d12d7fd4f505a1fa66ad3c465f6d3621dda79887af91fa9a2fa273b689fa1fdfENV
TORCH_CUDNN_V8_API_ENABLED=1
01/25/2024 5:13 AM UTC
b495cb014b8c0c5815ddef4c8d542edb2b8222be7f03f6658da0f3b0837e7f11RUN
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
01/25/2024 5:13 AM UTC
...