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
de1638c0fd16a6c26d7ae924fbb2befa69f5fc32bd0e1dc03eca4bf3a57b1ae5CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
07/09/2024 1:36 AM UTC
b68f1c3876222b6e1e55c47ef049d3331fdce57a83e7157cd500e7f06b948b97ARG
NVIDIA_BUILD_REF=df5ce920d3adddfc694c4a4e440f2bd72a7c00f0
07/09/2024 1:36 AM UTC
c9e8ea1bba8bd27ca390107f457c5da3aca09798298dcebaad0527977949fa92LABEL
com.nvidia.build.id=100464919
07/09/2024 1:36 AM UTC
01e2ef52c91b9b22d18eb761d4804fb5403522c34688670e3ee9d240ef4e0087ENV
NVIDIA_BUILD_ID=100464919
07/09/2024 1:36 AM UTC
ca974ed81f777432dc3452a24e3d0228ab4fab9febac63ed43dcb9c78e85e9b5ARG
NVIDIA_BUILD_ID=100464919
07/09/2024 1:36 AM UTC
6ef97ce7f2273c44dc3d4fcef08449aea082546d5543925d2d551aaedb39d55dCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
07/09/2024 1:36 AM UTC
84e6ae11e20c9009468896f5093748055c5437085fd52762ab96c29f91d61089RUN
NVIDIA_PYTORCH_VERSION=24.07 PYTORCH_BUILD_VERSION=2.4.0a0+3bcc3cd NVFUSER_BUILD_VERSION=f73ff1b 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
07/09/2024 1:36 AM UTC
d391e0bc5c35c5199a73de24e09a2f46eaaad689e619ad51dd044f94c5660fa8ENV
CUDA_MODULE_LOADING=LAZY
07/09/2024 1:36 AM UTC
e441122a4e04ffb7b5109038a012c00a4c09e30591d754dc79739658bc54741eENV
TORCH_CUDNN_V8_API_ENABLED=1
07/09/2024 1:36 AM UTC
3d2adaa077d14aeda3ffeba959779652d7b2cda19c85c445fbb2668ded3892d7RUN
NVIDIA_PYTORCH_VERSION=24.07 PYTORCH_BUILD_VERSION=2.4.0a0+3bcc3cd NVFUSER_BUILD_VERSION=f73ff1b 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 pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}; fi
07/09/2024 1:36 AM 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.