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.

LayerLabelCreated
257075c334fb826ccf1ec1a3cedba4f835d4badd4e2c710b6840a9c796d236b4CONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
02/18/2021 5:42 AM UTC
da5a8201291133120cb8058aee9aab627a8f7ad10a73146d78ede6554a78f526ARG
NVIDIA_BUILD_REF
02/18/2021 5:42 AM UTC
61484f00ae49c339f8be5876eafd3449834eba8f436d2fea2488c6286a797944LABEL
com.nvidia.build.id=20138038
02/18/2021 5:42 AM UTC
bf9916dc0a86814ee29460dc1775a922d2e8cc588d63f8c3bc1ef69494c5780dENV
NVIDIA_BUILD_ID=20138038
02/18/2021 5:42 AM UTC
47b32a364f61d3f194d0790d2c94dab59375e5d28ea47ecfb4b2b51609d47cb0ARG
NVIDIA_BUILD_ID
02/18/2021 5:42 AM UTC
5d8239704d0019ef426d7066214ebb7489326ab05a57adf8d32369625de0a80aRUN
NVIDIA_PYTORCH_VERSION=21.02 PYTORCH_BUILD_VERSION=1.8.0a0+52ea372 PYVER=3.8 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
02/18/2021 5:42 AM UTC
3274c0e90815fc8c85e529d7adf116224012e767c1dadcb879852eb166eed4daENV
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
02/18/2021 5:42 AM UTC
2ade1f65960df80b04f92087c9f6ebedf583b8c801f9b48942d1d1e3470dbf32RUN
NVIDIA_PYTORCH_VERSION=21.02 PYTORCH_BUILD_VERSION=1.8.0a0+52ea372 PYVER=3.8 URL=$(VERIFY=1 /nvidia/build-scripts/installTRT.sh 2>/dev/null | sed -n "s/^.*\(http.*\)deb.*$/\1/p")tar &&
  FILE=$(wget -O - $URL 2>/dev/null | sed -n 's/^.*href="\(TensorRT[^"]*\)".*$/\1/p' | grep -v internal) &&
  wget --quiet $URL/$FILE -O - | tar -xz &&
  PY=$(python -c 'import sys; print(str(sys.version_info[0])+str(sys.version_info[1]))') &&
  pip install TensorRT-*/python/tensorrt-*-cp$PY*.whl &&
  pip install TensorRT-*/graphsurgeon/graphsurgeon-*.whl &&
  pip install TensorRT-*/uff/uff-*.whl &&
  mv /usr/src/tensorrt /opt &&
  ln -s /opt/tensorrt /usr/src/tensorrt &&
  rm -r TensorRT-* &&
  UFF_PATH=$(pip show uff | sed -n 's/Location: \(.*\)$/\1/p')/uff &&
  sed -i 's/from tensorflow import GraphDef/from tensorflow.python import GraphDef/' $UFF_PATH/converters/tensorflow/conversion_helpers.py &&
  chmod +x ${UFF_PATH}/bin/convert_to_uff.py &&
  ln -sf ${UFF_PATH}/bin/convert_to_uff.py /usr/local/bin/convert-to-uff
02/18/2021 5:42 AM UTC
57e65048dd6337807df346b4de14416edcb00fea0b4a605c347827a416d1f38bCOPY
NVIDIA_Deep_Learning_Container_License.pdf /workspace/
02/18/2021 5:42 AM UTC
240a59631659a88df552d6085acae6be03bbd1cecfb0ffc08b0a80ff425fb1f0ENTRYPOINT
/usr/local/bin/nvidia_entrypoint.sh
02/18/2021 5:42 AM 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.