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
57b1202ca6be5130da3ec4842ab22c1c493e6594fe11d846702d8c68e11ceb65CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
10/03/2024 9:46 PM UTC
4f0faedde4ff3685392001c3084ca3ba5f44b53a2da81f1760a3b4e366069fddARG
NVIDIA_BUILD_REF=3e3c067dd015e6d16d2cf59ac18e9f2e2466b68a
10/03/2024 9:46 PM UTC
3f46f687ab69729198ff09ef4d242798dcc91370e05421b133abf8f8f64d3dc8LABEL
com.nvidia.build.id=114410972
10/03/2024 9:46 PM UTC
1f4515d7925e983952ea5236676f29e2f7c7a9e3a2690dcd907a060bda8b10a2ENV
NVIDIA_BUILD_ID=114410972
10/03/2024 9:46 PM UTC
95ca1443f44df1e6a2c905f828c8e56c318a17036a4629c76a4454868e6f1585ARG
NVIDIA_BUILD_ID=114410972
10/03/2024 9:46 PM UTC
a84dbb1de4d8e58476da4821a5530bbead27275ec2fb30d6e8a16cddeeeb7f8dCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
10/03/2024 9:46 PM UTC
03efb8e5acb35cdc0ee96fb6b0242827aa266ec119079becaa88953c6611ca72RUN
NVIDIA_PYTORCH_VERSION=24.10 PYTORCH_BUILD_VERSION=2.5.0a0+e000cf0 NVFUSER_BUILD_VERSION=f669fcf TARGETARCH=amd64 PYVER=3.10 L4T=0 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
10/03/2024 9:46 PM UTC
40fd17f6e08b714d22f3c823070ed62454995f9e0872c99e1c5590882948dec4ENV
CUDA_MODULE_LOADING=LAZY
10/03/2024 9:46 PM UTC
853950d6e9e8df7f6244252dacd27f711b24c5d5b1a81d3737095498102603edENV
TORCH_CUDNN_V8_API_ENABLED=1
10/03/2024 9:46 PM UTC
049b0fa3a8ff910705ecde187173ddc41f9417c07ea54f7d364b9e5d3b629e74RUN
NVIDIA_PYTORCH_VERSION=24.10 PYTORCH_BUILD_VERSION=2.5.0a0+e000cf0 NVFUSER_BUILD_VERSION=f669fcf TARGETARCH=amd64 PYVER=3.10 L4T=0 PYVER=3.10 if [ "${L4T}" = "1" ]; then echo "Not installing Transformer Engine in iGPU container until Version variable is set"; else NVTE_BUILD_THREADS_PER_JOB=8 pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}; fi
10/03/2024 9:46 PM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.