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
7d96f88d1dbb8a83cea10cb23b13cc2015e2326c24e17a0f314bd540c3c9a67bCONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
11/05/2020 8:00 PM UTC
fa025ed1ca853e62066fd6d85039d87ac042d68f240a179906d2d82cd1a657aaARG
NVIDIA_BUILD_REF
11/05/2020 8:00 PM UTC
76466d226f1c779f39c01d3a40eb8893a3dd9624996815b7d8cb1763741b0582LABEL
com.nvidia.build.id=17345815
11/05/2020 8:00 PM UTC
8ca57006cc3bf7967223ecee7412f15f41a8fc8324c82c6377a153a864c46a14ENV
NVIDIA_BUILD_ID=17345815
11/05/2020 8:00 PM UTC
5de28334318cdef297b5ca821f691207367abe4c9500ab1c56ac150e49b8a579ARG
NVIDIA_BUILD_ID
11/05/2020 8:00 PM UTC
cbfe5799b4bffc9906ecc50de73f52bf360d29fba6e37fe1232e2b41310485a5RUN
NVIDIA_PYTORCH_VERSION=20.11 PYTORCH_BUILD_VERSION=1.8.0a0+17f8c32 PYVER=3.6 pip install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com --extra-index-url https://tensorrt-read-only:Tensorrt\@123@urm.nvidia.com/artifactory/api/pypi/sw-tensorrt-pypi/simple polygraphy
11/05/2020 8:00 PM UTC
77a45531213b1d9f972a98d3d96eff93f57e71906d6ec2d48149efb58c6062e6ENV
PATH=/opt/conda/bin:/opt/cmake-3.14.6-Linux-x86_64/bin/:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin
11/05/2020 8:00 PM UTC
5a40a6f7155ddaab2fc65779e81e01beb93afd4dd939e6e691910df651877f31RUN
NVIDIA_PYTORCH_VERSION=20.11 PYTORCH_BUILD_VERSION=1.8.0a0+17f8c32 PYVER=3.6 BASE=0 DEVEL=0 SAMPLES=0 PYTHON=1 QAT=1 /nvidia/build-scripts/installTRT.sh &&
  mv /usr/src/tensorrt /opt &&
  ln -s /opt/tensorrt /usr/src/tensorrt
11/05/2020 8:00 PM UTC
32e091ee8f7e7000f36a2a6c41ac6e0be7d2a1add7a455578c474071c4ec7e86ENTRYPOINT
/usr/local/bin/nvidia_entrypoint.sh
11/05/2020 8:00 PM UTC
10257896513c26d46f7f69c63b318dc3bb5617166ff98690dd2c65ece1130bedCOPY
nvidia_entrypoint.sh /usr/local/bin
11/05/2020 8:00 PM 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.