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
0a14f5d947b501883666e1f251834a2d389ac58d35de829afba1245ccda741ddCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
10/13/2022 9:36 PM UTC
7b09830c7ed1554e630130ceb0bfe8c0a0e95e0ef4c6580c557c8a6ba37b2dadARG
NVIDIA_BUILD_REF
10/13/2022 9:36 PM UTC
706486bf9e01c416e38110cff59bc570427d4e00b1fa4ed49f112963371fae36LABEL
com.nvidia.build.id=46164382
10/13/2022 9:36 PM UTC
29637781bf55bea8ad8be52653ace6e56931e43df8025438517bf0a5f8cd3b82ENV
NVIDIA_BUILD_ID=46164382
10/13/2022 9:36 PM UTC
f748728cf12bbe2aba829c6199f89d3078398d1827eabd60b5772bd913884363ARG
NVIDIA_BUILD_ID
10/13/2022 9:36 PM UTC
553eaeb1d7fe50daa1cb6d2f7e05cd55aa36c61bab1935f4dc98f01cef6e5372COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
10/13/2022 9:36 PM UTC
0e88d3d33aa5fc658cc23369b222aee8dd62cdde0257966b2fe2c0bd5765f667RUN
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
10/13/2022 9:36 PM UTC
e6ed2d9e799575acee63967bfd1939f31f35caf1e482f5c8b291ecbe1e89cf8eENV
CUDA_MODULE_LOADING=LAZY
10/13/2022 9:36 PM UTC
24c0e19605a07ad622a7cfd43821c58e0c91f65c27d145fedb1671d08a79dcffENV
TORCH_CUDNN_V8_API_ENABLED=1
10/13/2022 9:36 PM UTC
cca24289171dac65e329ea38b5181d75779204004b546de82112aaf7786bbf7bRUN
PYVER=3.8 pip install --no-cache-dir /tmp/te_wheel/*.whl
10/13/2022 9:36 PM UTC
...