Open ys7yoo opened 5 years ago
tried
CC=/usr/bin/gcc-5 CXX=/usr/bin/g++-5 CUDA_HOST_COMPILER=/usr/bin/gcc-5 tools/build_pytorch_libs.sh --use-cuda --use-nnpack --use-qnnpack caffe2
Testing with
rm CMakeCache.txt
CC=/usr/bin/gcc-5 CXX=/usr/bin/g++-5 CUDA_HOST_COMPILER=/usr/bin/gcc-5 CUDNN_INCLUDE_DIR=/usr/local/cuda/include CUDNN_LIBRARY=/usr/lib/x86_64-linux-gnu tools/build_pytorch_libs.sh --use-cuda --use-nnpack --use-mkldnn --use-qnnpack caffe2