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

Docker::Installers: shfmt tensorrt installer

上级 4b89f7ad
...@@ -73,7 +73,7 @@ fixed_version=true ...@@ -73,7 +73,7 @@ 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}" \
...@@ -84,7 +84,7 @@ if ${fixed_version}; then ...@@ -84,7 +84,7 @@ if ${fixed_version}; then
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 \
...@@ -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.
先完成此消息的编辑!
想要评论请 注册