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
6ef7d1ede02185c7876ba1219bdd7f788bada7785468a6468110ae1effefc398CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
05/03/2024 4:13 AM UTC
ced8904254747d3ef188d868f0c7fb9b555a4afaace5b37e85c195b489d5aab4ARG
NVIDIA_BUILD_REF
05/03/2024 4:13 AM UTC
caee56691ad3ca79d9087ddb53d9149d15f04b1ba6e6d6209309c943b6ed9b55LABEL
com.nvidia.build.id=91431255
05/03/2024 4:13 AM UTC
b4a0c43c677e4c99253acdfd660505aad2989e08a673b2e6db8694ff9a633cbfENV
NVIDIA_BUILD_ID=91431255
05/03/2024 4:13 AM UTC
06984e77b51bd155a79e48745c549f741017456f9fa1120b57252c4bc00686dbARG
NVIDIA_BUILD_ID
05/03/2024 4:13 AM UTC
1c5bc68f0791c3e12bc0bcaa3a9e7b7120d4d9ec4ef95dc63607212c36016cbbCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
05/03/2024 4:13 AM UTC
bb1ad49a8794072580de408eae793dadf2465fa1f9741d5ecfa3f62cc28990f9RUN
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
05/03/2024 4:13 AM UTC
ee0c825ee8b81c90df10821bda6d5ea2a31cf5d84ae497ccf637afa7262de0b8ENV
CUDA_MODULE_LOADING=LAZY
05/03/2024 4:13 AM UTC
35fe60e1c9be31f83102cf7e638e967bae276739a17219aea9efd823e96a0937ENV
TORCH_CUDNN_V8_API_ENABLED=1
05/03/2024 4:13 AM UTC
d797075f877d9dbaab8352d8f5c97195154b08ca5a8aa7d7a8fec18578b87f83RUN
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
05/03/2024 4:13 AM UTC
...