marcoabreu commented on a change in pull request #19011:
URL: https://github.com/apache/incubator-mxnet/pull/19011#discussion_r476750212
##########
File path: ci/docker/runtime_functions.sh
##########
@@ -1069,6 +1069,22 @@ unittest_ubuntu_python3_gpu_nocudnn() {
nosetests-3.4 $NOSE_COVERAGE_ARGUMENTS $NOSE_TIMER_ARGUMENTS --with-xunit
--xunit-file nosetests_gpu.xml --verbose tests/python/gpu
}
+unittest_ubuntu_tensorrt_gpu() {
+ set -ex
+ export PYTHONPATH=./python/
+ export MXNET_STORAGE_FALLBACK_LOG_VERBOSE=0
+ export MXNET_SUBGRAPH_VERBOSE=0
+ export LD_LIBRARY_PATH=/work/mxnet/lib:$LD_LIBRARY_PATH
+ export CUDNN_VERSION=${CUDNN_VERSION:-7.0.3}
+ export MXNET_ENABLE_CYTHON=0
+ export DMLC_LOG_STACK_TRACE_DEPTH=10
+ pip3 install --extra-index-url
https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda100==0.24
Review comment:
@leezu could you assist reviewing this PR? I think the structure wrt
installation of packages and creating a new job might not be in line.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
