提交 535c8673 编写于 作者: S storypku 提交者: Liu Jiaming

Docker::Installers: shfmt tensorrt installer

上级 4b89f7ad
...@@ -73,27 +73,27 @@ fixed_version=true ...@@ -73,27 +73,27 @@ fixed_version=true
if ${fixed_version}; then if ${fixed_version}; then
VERSION="7.0.0-1+cuda10.2" VERSION="7.0.0-1+cuda10.2"
apt-get -y update && \ apt-get -y update &&
apt-get -y install \ apt-get -y install \
libnvinfer7="${VERSION}" \ libnvinfer7="${VERSION}" \
libnvonnxparsers7="${VERSION}" \ libnvonnxparsers7="${VERSION}" \
libnvparsers7="${VERSION}" \ libnvparsers7="${VERSION}" \
libnvinfer-plugin7="${VERSION}" \ libnvinfer-plugin7="${VERSION}" \
libnvinfer-dev="${VERSION}" \ libnvinfer-dev="${VERSION}" \
libnvonnxparsers-dev="${VERSION}" \ libnvonnxparsers-dev="${VERSION}" \
libnvparsers-dev="${VERSION}" \ libnvparsers-dev="${VERSION}" \
libnvinfer-plugin-dev="${VERSION}" libnvinfer-plugin-dev="${VERSION}"
else else
apt-get -y update && \ apt-get -y update &&
apt-get -y install \ apt-get -y install \
libnvinfer7 \ libnvinfer7 \
libnvonnxparsers7 \ libnvonnxparsers7 \
libnvparsers7 \ libnvparsers7 \
libnvinfer-plugin7 \ libnvinfer-plugin7 \
libnvinfer-dev \ libnvinfer-dev \
libnvonnxparsers-dev \ libnvonnxparsers-dev \
libnvparsers-dev \ libnvparsers-dev \
libnvinfer-plugin-dev libnvinfer-plugin-dev
fi fi
# FIXME(all): # FIXME(all):
...@@ -104,7 +104,7 @@ fi ...@@ -104,7 +104,7 @@ fi
# 2) move to cudnn installer section # 2) move to cudnn installer section
CUDNN_HEADER_DIR="/usr/include/$(uname -m)-linux-gnu" CUDNN_HEADER_DIR="/usr/include/$(uname -m)-linux-gnu"
[ -e "${CUDNN_HEADER_DIR}/cudnn.h" ] || \ [ -e "${CUDNN_HEADER_DIR}/cudnn.h" ] ||
ln -s "${CUDNN_HEADER_DIR}/cudnn_v7.h" "${CUDNN_HEADER_DIR}/cudnn.h" ln -s "${CUDNN_HEADER_DIR}/cudnn_v7.h" "${CUDNN_HEADER_DIR}/cudnn.h"
# Disable nvidia apt sources.list settings to speed up build process # Disable nvidia apt sources.list settings to speed up build process
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册