Caffe Environment Config

# Install Caffe ```sh RUN git clone https://github.com/yubaoliu/caffe.git \ && cd caffe \ && mkdir build \ && cd build \ && cmake -DUSE_CUDNN=1 -DUSE_NCCL=1 .. \ && make all \ && make install ``` # Possible Errors ## CUDA_cublas_device_LIBRARY ```sh 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_device_LIBRARY (ADVANCED) linked by target "caffe" in directory ``` Solution: Update Cmake version ## cudnn.hpp(112): error: too few arguments in function call ```sh #warning "math_functions.h is an internal header file and must not be used directly. This file will be removed in a future CUDA release. Please use cuda_runtime_api.h or cuda_runtime.h instead." ^~~~~~~ /root/caffe-segnet-cudnn5/include/caffe/util/cudnn.hpp(112): error: too few arguments in function call 1 error detected in the compilation of "/tmp/tmpxft_000010e5_00000000-4_math_functions.cpp4.ii". CMake Error at cuda_compile_1_generated_math_functions.cu.o.Release.cmake:280 (message): Error generating file /root/caffe-segnet-cudnn5/build/src/caffe/CMakeFiles/cuda_compile_1.dir/util/./cuda_compile_1_generated_math_functions.cu.o src/caffe/CMakeFiles/caffe.dir/build.make:502: recipe for target 'src/caffe/CMakeFiles/cuda_compile_1.dir/util/cuda_compile_1_generated_math_functions.cu.o' failed make[2]: *** [src/caffe/CMakeFiles/cuda_compile_1.dir/util/cuda_compile_1_generated_math_functions.cu.o] Error 1 CMakeFiles/Makefile2:402: recipe for target 'src/caffe/CMakeFiles/caffe.dir/all' failed make[1]: *** [src/caffe/CMakeFiles/caffe.dir/all] Error 2 Makefile:148: recipe for target 'all' failed make: *** [all] Error 2 ``` Solution: Seee https://github.com/BVLC/caffe/issues/5793 # Reference - https://github.com/NVIDIA/DIGITS/issues/2185 - Install Cmake: https://cmake.org/download/