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
57121fde0c94f73423f3f4c70d5d0b5379912f8b8e52dadbca6e1bca7b248334CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
07/06/2023 10:07 PM UTC
6204921c9d6493a8a488b09eb8ed6811cec2cfec69f217fce903c467ca46c56dARG
NVIDIA_BUILD_REF
07/06/2023 10:07 PM UTC
a12788aae3b5b5930b81c5b45ee6d41d9b44b5ae4b10cc904a754b224018c59eLABEL
com.nvidia.build.id=63867923
07/06/2023 10:07 PM UTC
b82414515778049f32e174684303c1ce6611cb143163262a4981c399c719ef45ENV
NVIDIA_BUILD_ID=63867923
07/06/2023 10:07 PM UTC
4c1cd0adfb6ff735aa18c0283b29916bda773361b42edd2e9879d539436d6309ARG
NVIDIA_BUILD_ID
07/06/2023 10:07 PM UTC
1f4fdfea45383c6cff9ab69e0984fa8b3d7ac9706bfd8d83545c220a99ca1eaaCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
07/06/2023 10:07 PM UTC
d6ce06900fe4ebe205c26065ebe465a748095c5c811727f34d379567e2a35871RUN
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
07/06/2023 10:07 PM UTC
ac0ca41ccd4eaab13fc6ed65f2502537ad2822cec11ac01b83e3e606ffe73914ENV
CUDA_MODULE_LOADING=LAZY
07/06/2023 10:07 PM UTC
b205089956578174aecffa6d1d3f99f4c5b8a05c755885629bf44aece4b39365ENV
TORCH_CUDNN_V8_API_ENABLED=1
07/06/2023 10:07 PM UTC
e91e068fda8018315a622f35445957a77fa790d4c6fbab474b98788b3270c11fRUN
PYVER=3.10 pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}
07/06/2023 10:07 PM UTC
...