Hi, all. I want to compile sten alone, because I want to use pytorch in C++. when I follow the install guide in https://github.com/pytorch/pytorch/tree/master/aten , I got an error.
I just use cmake ..
, and the error is :
-- Found system Eigen at /usr/local/include/eigen3
-- Could NOT find pybind11 (missing: pybind11_INCLUDE_DIR)
-- Found CUDA: /usr/local/cuda (found suitable version "9.0", minimum required is "7.0")
-- Caffe2: CUDA detected: 9.0
-- Caffe2: CUDA nvcc is: /usr/local/cuda/bin/nvcc
-- Caffe2: CUDA toolkit directory: /usr/local/cuda
-- Caffe2: Header version is: 9.0
-- Found cuDNN: v7.1.4 (include: /usr/local/cuda/include, library: /usr/local/cuda/lib64/libcudnn.so)
-- Autodetected CUDA architecture(s): 6.1
-- Added CUDA NVCC flags for: -gencode;arch=compute_61,code=sm_61
CMake Error at /home/mjj/Pytorch/pytorch/cmake/public/utils.cmake:5 (set):
set given invalid arguments for CACHE mode.
Call Stack (most recent call first):
/home/mjj/Pytorch/pytorch/cmake/Dependencies.cmake:480 (caffe2_update_option)
CMakeLists.txt:65 (include)
anyone could help me ? thx very much!!