PyTorch | NVIDIA NGC

NVIDIA

PyTorch

Container

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

Layer	Label		Created
c44b907487357d6d83d4412291cc5faa1105bd5da2ab65bedb95943d6db99a40	CONFIG	`Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp`	04/15/2023 2:28 AM UTC
4fe73b5bdde8021639ac76e30b4634a15901a53598183673b03b725509a755b0	ARG	`NVIDIA_BUILD_REF`	04/15/2023 2:28 AM UTC
ad520c17da8f94de3007fe61a7ed6987102750e18528c94d7e14eeca02ae35f6	LABEL	`com.nvidia.build.id=58180998`	04/15/2023 2:28 AM UTC
799cb57635d9c787977c20a2fc11e7fe7f2ef5695ab7387d010d880f2b2fbb5f	ENV	`NVIDIA_BUILD_ID=58180998`	04/15/2023 2:28 AM UTC
5160e16f9f8a13dd78aafb43953deeecd3962543cc23bc0a13cbf87ed30bfe1e	ARG	`NVIDIA_BUILD_ID`	04/15/2023 2:28 AM UTC
b70619e6f06634ee11b890c0ecbdca957fdc643377acb1c0035de6d1f425a353	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	04/15/2023 2:28 AM UTC
f592532993f6cc05dda2e0bbc05b30e42299053fce6313b6ab1500abf35cb6fa	RUN	`PYVER=3.8 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/15/2023 2:28 AM UTC
ca5e3c13b7e4cdd8de7db4d21a4bbcc25437cd1ac91308a57a4e335866f90eaf	ENV	`CUDA_MODULE_LOADING=LAZY`	04/15/2023 2:28 AM UTC
e1da74c478852db2da208d00e90e157f27b06b334278df21d058cfc785f5a52a	ENV	`TORCH_CUDNN_V8_API_ENABLED=1`	04/15/2023 2:28 AM UTC
362d42f1e3bf35eb253e07d9947d974a48d73e8cd229c88d0bb63628bba6a1d7	RUN	`PYVER=3.8 pip install --no-cache-dir git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}`	04/15/2023 2:28 AM UTC