Error buidl source

[sonfire186]

docker image nvidia

-- The CXX compiler identification is GNU 7.4.0
-- The CUDA compiler identification is NVIDIA 10.0.130
-- Check for working CXX compiler: /usr/bin/c++
-- Check for working CXX compiler: /usr/bin/c++ -- works
-- Detecting CXX compiler ABI info
-- Detecting CXX compiler ABI info - done
-- Detecting CXX compile features
-- Detecting CXX compile features - done
-- Check for working CUDA compiler: /usr/local/cuda/bin/nvcc
-- Check for working CUDA compiler: /usr/local/cuda/bin/nvcc -- works
-- Detecting CUDA compiler ABI info
-- Detecting CUDA compiler ABI info - done
-- Found Protobuf: /usr/local/lib/libprotobuf.so;-lpthread (found version "3.6.1")
-- PROTOBUF=/usr/local/lib/libprotobuf.so
-- Performing Test SUPPORT_CXX11
-- Performing Test SUPPORT_CXX11 - Success
-- Found Threads: TRUE
-- Found CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda
-- Found CUDA_CUDA_LIBRARY=/usr/local/cuda/targets/x86_64-linux/lib/stubs/libcuda.so
-- Found CUDA_CUDART_LIBRARY=/usr/local/cuda/lib64/libcudart.so
-- Found CUDA_NVRTC_LIBRARY=CUDA_NVRTC_LIBRARY-NOTFOUND
-- Found CUDA_CUDNN_LIBRARY=CUDA_CUDNN_LIBRARY-NOTFOUND
-- Found CUDA_CUBLAS_LIBRARY=CUDA_CUBLAS_LIBRARY-NOTFOUND
-- CUDA_INCLUDE_DIR=/usr/local/cuda/include
CMake Error: The following variables are used in this project, but they are set to NOTFOUND.
Please set them or make sure they are set and tested correctly in the CMake files:
CUDA_CUBLAS_LIBRARY
    linked by target "taso_runtime" in directory /storage/data/taso
CUDA_CUDNN_LIBRARY
    linked by target "taso_runtime" in directory /storage/data/taso

-- Configuring incomplete, errors occurred!
See also "/storage/data/taso/build/CMakeFiles/CMakeOutput.log".
See also "/storage/data/taso/build/CMakeFiles/CMakeError.log".

[sonfire186]

TF GPU it`s work

[jiazhihao]

@sonfiree are you using the pre-built docker image? Make sure you have installed nvidia-docker, and launch the TASO docker image via run_docker.sh tasoml/cuda100.

[sonfire186]

I am using nvidia-docker. I use Build TASO Runtime or Docker Images?

[jiazhihao]

You don't need to build the docker image.

Once you have cloned the TASO github repo, you can directly launch our prebuilt docker image via /path/to/taso/docker/run_docker.sh tasoml/cuda100.

[sonfire186]

I have Nvidia-Docker running and I need to run TASO. I do not need to create another container

[jiazhihao]

@sonfiree Just to double check that we are on the same page: when you got the above error messages, were you trying to build a TASO docker image through docker build, or were you trying to run our pre-built docker using the run_docker script.

[sonfire186]

I'm already in the container.

[jiazhihao]

It seems cmake cannot automatically find CUDA libraries to build the runtime. Can you change set(USE_CUDA on) to set(USE_CUDA /usr/local/cuda/) and also make sure libcublas.so and libcudnn.so are in the folder /usr/local/cuda/lib64? They should be there in our docker image.