You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

3 lines
160 B

cd build
cmake .. -DCMAKE_PREFIX_PATH=../../libtorch/share/cmake/Torch -DCUDNN_LIBRARY_PATH=../cuda/lib64/libcudnn.so -DCUDNN_INCLUDE_PATH=../cuda/include
make