Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Fixed and re-enables TensorRT steps #14960

Merged
merged 3 commits into from
May 15, 2019
Merged

Fixed and re-enables TensorRT steps #14960

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
68a9771
Pins libnvinfer versions
perdasilva May 15, 2019
3b03334
Sets cudnn to version 7.5.0 in tensorrt environment
perdasilva May 15, 2019
1ba8fa6
Re-enables TensorRT stages
perdasilva May 15, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
#
# Dockerfile to run MXNet on Ubuntu 16.04 for CPU

FROM nvidia/cuda:10.0-cudnn7-devel
FROM nvidia/cuda:10.0-devel

WORKDIR /work/deps

Expand All @@ -35,6 +35,10 @@ ARG USER_ID=0
COPY install/ubuntu_adduser.sh /work/
RUN /work/ubuntu_adduser.sh

ENV CUDNN_VERSION=7.5.0.56
COPY install/ubuntu_cudnn.sh /work/
RUN /work/ubuntu_cudnn.sh

COPY runtime_functions.sh /work/

WORKDIR /work/mxnet
Expand Down
3 changes: 2 additions & 1 deletion ci/docker/install/tensorrt.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,5 +44,6 @@ echo "TensorRT build enabled. Installing TensorRT."
wget -qO tensorrt.deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda10.0_1-1_amd64.deb
dpkg -i tensorrt.deb
apt-get update
apt-get install -y --allow-downgrades libnvinfer-dev
apt-get install -y --allow-downgrades libnvinfer5=5.1.2-1+cuda10.0
apt-get install -y --allow-downgrades libnvinfer-dev=5.1.2-1+cuda10.0
rm tensorrt.deb
2 changes: 1 addition & 1 deletion ci/docker/install/ubuntu_cudnn.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -50,5 +50,5 @@ case ${CUDA_VERSION} in
;;
esac

apt-get install -y libcudnn7=${libcudnn7_version} libcudnn7-dev=${libcudnn7_dev_version}
apt-get install -y --allow-downgrades libcudnn7=${libcudnn7_version} libcudnn7-dev=${libcudnn7_dev_version}

4 changes: 2 additions & 2 deletions ci/jenkins/Jenkinsfile_unix_gpu
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ core_logic: {
custom_steps.compile_unix_full_gpu(),
custom_steps.compile_unix_cmake_mkldnn_gpu(),
custom_steps.compile_unix_cmake_gpu(),
// custom_steps.compile_unix_tensorrt_gpu(),
custom_steps.compile_unix_tensorrt_gpu(),
custom_steps.compile_unix_int64_gpu()
])

Expand All @@ -51,7 +51,7 @@ core_logic: {
custom_steps.test_unix_python2_mkldnn_gpu(),
custom_steps.test_unix_python3_mkldnn_gpu(),
custom_steps.test_unix_python3_mkldnn_nocudnn_gpu(),
// custom_steps.test_unix_python3_tensorrt_gpu(),
custom_steps.test_unix_python3_tensorrt_gpu(),
custom_steps.test_unix_perl_gpu(),
custom_steps.test_unix_r_gpu(),
custom_steps.test_unix_cpp_gpu(),
Expand Down