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
da9d5a89fd4d929efbd4f47c4fd03f475145f51c7a5e3693b503fc1b71812c9dCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
04/05/2024 7:54 PM UTC
04ea05b2fb543c6f074eb881e609a24c3d3a8594ed02596dc645e519662e372aARG
NVIDIA_BUILD_REF
04/05/2024 7:54 PM UTC
946c78794a17e4d034898ddc9d97edb8245375e2c168447ee4f3022904d15564LABEL
com.nvidia.build.id=88113656
04/05/2024 7:54 PM UTC
b1ba48a810bba951f7c8cb697047a052801e15a82a3c38b3134a75c8e89cb176ENV
NVIDIA_BUILD_ID=88113656
04/05/2024 7:54 PM UTC
41ab94373f97c98bb789076969780bd566097728c31aca7e3281ebfb09247b47ARG
NVIDIA_BUILD_ID
04/05/2024 7:54 PM UTC
4efca7315837f54c2a50fd22da51810e91baaa1756c7e143d961f9ea60b30780COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
04/05/2024 7:54 PM UTC
992523500d35b3a943f366dd795229f948a1013f85312d8140da39e3cfe75b7eRUN
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
04/05/2024 7:54 PM UTC
9a0da99b0695076f99f397e4899634fab33e50f7b7d363d63bc5df9a4defa0ebENV
CUDA_MODULE_LOADING=LAZY
04/05/2024 7:54 PM UTC
bb93860f2928005827b9c1cb54c93c5ff751dacd07b4003938f9e937ee9e9e22ENV
NCCL_WORK_FIFO_DEPTH=4194304
04/05/2024 7:54 PM UTC
cff22df93407b4fe1245a5e1ab114ddef315a2e7aec896e57a80afea80f9be36ENV
TORCH_CUDNN_V8_API_ENABLED=1
04/05/2024 7:54 PM UTC
...