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
b2637bbdf5fb584a064cb88a1238ec8fe91d9ace4e7872e7902e7fe554a4aa15CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
08/07/2023 12:21 AM UTC
a652543e6ffe05d557ae745a9f34e99cd6d737e130ca72d3e856cfcec18d482eARG
NVIDIA_BUILD_REF
08/07/2023 12:21 AM UTC
06a2385a29154c6ab6df23a868b875ddcfa12222727a93aa1983c2bc8837265fLABEL
com.nvidia.build.id=66128610
08/07/2023 12:21 AM UTC
13a97a5bcffc8b3ae92a3b4255aeb8f1e8eadc52b8b1ac7a0ae006cc5384af4dENV
NVIDIA_BUILD_ID=66128610
08/07/2023 12:21 AM UTC
d391bae8ebd1b377bc46ec49b39544712a4a750845a25e47a509f3cf78a25d08ARG
NVIDIA_BUILD_ID
08/07/2023 12:21 AM UTC
4fff5d500ba1323334a061f5458774dc9458db164a475c186a57adb09d5f1e5eCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
08/07/2023 12:21 AM UTC
7b9cfa3f652889bca95e023f4eedee87f9225d25a4ce15bef2ff98aaa74dde62RUN
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
08/07/2023 12:21 AM UTC
a30e9cb88ebb0fb1bf6560cd1242dba29d20857a337e25c65281be7011b0976fENV
CUDA_MODULE_LOADING=LAZY
08/07/2023 12:21 AM UTC
04873515577495b068fd285d968e572b093e73ed72ac3ad5f23cb77845484032ENV
TORCH_CUDNN_V8_API_ENABLED=1
08/07/2023 12:21 AM UTC
120ede3fed30ed9d3521de7e2b8c4b5b6e349c95c115def1e29216092ec287ddRUN
PYVER=3.10 env MAX_JOBS=1 pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}
08/07/2023 12:21 AM UTC
...