The hardcoding of CUDNN_VERSION in ./ci/docker/runtime_functions.sh is out-of-date.

DickJC123 commented 5 years ago

Description

Here's a portion of runtime_functions.sh relating to the tensorrt build (which is built against cuda 10.0 and cuDNN 7.4.2):

unittest_ubuntu_tensorrt_gpu() {
    set -ex
    export PYTHONPATH=./python/
    export MXNET_STORAGE_FALLBACK_LOG_VERBOSE=0
    export LD_LIBRARY_PATH=/work/mxnet/lib:$LD_LIBRARY_PATH
    export CUDNN_VERSION=7.0.3
    python tests/python/tensorrt/lenet5_train.py
    nosetests-3.4 $NOSE_COVERAGE_ARGUMENTS $NOSE_TIMER_ARGUMENTS --with-xunit --xunit-file nosetests_trt_gpu.xml --verbose --nocapture tests/python/tensorrt/
}

The runtime environment for other builds also set CUDNN_VERSION to 7.0.3, and those builds are against cuda 9.1 and cuDNN 7.1.2. It would be nice to have these env var settings fixed in a way that doesn't quickly become obsolete (or remove them if they are not needed).

mxnet-label-bot commented 5 years ago

Hey, this is the MXNet Label Bot. Thank you for submitting the issue! I will try and suggest some labels so that the appropriate MXNet community members can help resolve it. Here are my recommended labels: Cuda, Feature, CI

piyushghai commented 5 years ago

@mxnet-label-bot Add [CI]

@marcoabreu Maybe you can help with this in the runtime_functions.sh file ?

apache / mxnet

The hardcoding of CUDNN_VERSION in ./ci/docker/runtime_functions.sh is out-of-date. #14328

Description