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
08a60502c53271e228236332eb95063342d14809067dc52236287c5356c4aab2CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
11/21/2023 12:48 AM UTC
a8a56b9d67c0643590c40931f8a7c487b466a5e54638b871c7ae82a3722629f7ARG
NVIDIA_BUILD_REF
11/21/2023 12:48 AM UTC
8aa26034131ed8d2bea9166dd49d2d25ece43985d66eb74ae17776efbe6b4daaLABEL
com.nvidia.build.id=74957750
11/21/2023 12:48 AM UTC
c645db6a17f6a871f2b202100fbf9f1096b8851ac93803d42424fdae66d6a166ENV
NVIDIA_BUILD_ID=74957750
11/21/2023 12:48 AM UTC
3f9851f08e65299830ac7dd3810f5163c94d1254f34a635a57dcc1e903281410ARG
NVIDIA_BUILD_ID
11/21/2023 12:48 AM UTC
a25e0c38a3669e44b3b1668f0c14aa0d40dee49bad0d43d4ab976118eb8867c8COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
11/21/2023 12:48 AM UTC
bb3eaeb79198c19da4bb433d16d4ead8ec00cd29e772d2ec08d8ba5f81aa0241RUN
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
11/21/2023 12:48 AM UTC
9ac44154242a9936c446737d3740696d06d7b4a69ecee59d5338b05fab650fe9ENV
CUDA_MODULE_LOADING=LAZY
11/21/2023 12:48 AM UTC
60181fd30d27f2f160afcd04c4a32da1d474ae5e9aa6b6f86e29c82101aa08b1ENV
TORCH_CUDNN_V8_API_ENABLED=1
11/21/2023 12:48 AM UTC
70ee3fba3afe8ac09dc6b0652dbf891e51859c7260f3d916a4ebda3f757c1000RUN
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
11/21/2023 12:48 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.