Skip to content

Benchmarking script can pick CUDA dependency arbitrarily #14

Description

@psyhtest

TensorRT 3.0 only works with CUDA 9.0. However, I currently have both CUDA 9.0 and CUDA 9.1 installed:

$ ck show env --tags=cuda
Env UID: Target OS: Bits: Name: Version: Tags:
6f75374461b20f48 linux-64 64 cuDNN library 7.0.5 64bits,cuda,cudnn,dnn,host-os-linux-64,lib,target-os-linux-64,v7,v7.0,v7.0.5
ee992601ced4718a linux-64 64 Nvidia CUDA Compiler 9.1.85 64bits,compiler,cuda,host-os-linux-64,lang-c-cuda,lang-cpp-cuda,target-os-linux-64,v9,v9.1,v9.1.85
aa9bd57172aefafe linux-64 64 Nvidia CUDA Compiler 9.0.176 64bits,compiler,cuda,host-os-linux-64,lang-c-cuda,lang-cpp-cuda,target-os-linux-64,v9,v9.0,v9.0.176

The benchmarking script seems to arbitrarily pick either version:

$ grep cuda-compiler * -R -A2
bvlc-alexnet-tensorrt-3.0.1/pipeline.json: "cuda-compiler": {
bvlc-alexnet-tensorrt-3.0.1/pipeline.json- "sort": 20, bvlc-alexnet-tensorrt-3.0.1/pipeline.json- "detected_ver": "9.0.176", --
bvlc-googlenet-tensorrt-3.0.1/pipeline.json: "cuda-compiler": {
bvlc-googlenet-tensorrt-3.0.1/pipeline.json- "sort": 20, bvlc-googlenet-tensorrt-3.0.1/pipeline.json- "detected_ver": "9.1.85", --
deepscale-squeezenet-1.1-tensorrt-3.0.1/pipeline.json: "cuda-compiler": {
deepscale-squeezenet-1.1-tensorrt-3.0.1/pipeline.json- "sort": 20, deepscale-squeezenet-1.1-tensorrt-3.0.1/pipeline.json- "detected_ver": "9.0.176", 

When CK picks CUDA 9.1, compilation fails:

 "compile": {
"compilation_success": "no",
"compilation_success_bool": false,
"compilation_time": 0.8694801330566406,
"compilation_time_with_module": 2.111440896987915,
"fail_reason": "return code 1 !=0 ",
"joined_compiler_flags": "-O3"
}

A workaround is to temporarily "hide" CUDA 9.1 e.g. as follows:

$ ck show env --tags=cuda,v9.1
Env UID: Target OS: Bits: Name: Version: Tags:
ee992601ced4718a linux-64 64 Nvidia CUDA Compiler 9.1.85 64bits,compiler,cuda,host-os-linux-64,lang-c-cuda,lang-cpp-cuda,target-os-linux-64,v9,v9.1,v9.1.85
$ mv `ck find env:ee992601ced4718a`{,~}
$ ck show env --tags=cuda,v9.1

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions