зеркало из https://github.com/microsoft/caffe.git
24 строки
813 B
Makefile
24 строки
813 B
Makefile
# define third-party library paths
|
|
# CHANGE YOUR CUDA PATH IF IT IS NOT THIS
|
|
CUDA_DIR := /usr/local/cuda
|
|
# CHANGE YOUR CUDA ARCH IF IT IS NOT THIS
|
|
CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
|
|
-gencode arch=compute_20,code=sm_21 \
|
|
-gencode arch=compute_30,code=sm_30 \
|
|
-gencode arch=compute_35,code=sm_35
|
|
# CHANGE YOUR MKL PATH IF IT IS NOT THIS
|
|
MKL_DIR := /opt/intel/mkl
|
|
# CHANGE YOUR MATLAB PATH IF IT IS NOT THIS
|
|
# your mex binary should be located at $(MATLAB_DIR)/bin/mex
|
|
MATLAB_DIR := /usr/local
|
|
# PUT ALL OTHER INCLUDE AND LIB DIRECTORIES HERE
|
|
INCLUDE_DIRS := /usr/local/include /usr/include/python2.7 \
|
|
/usr/local/lib/python2.7/dist-packages/numpy/core/include
|
|
LIBRARY_DIRS := /usr/lib /usr/local/lib
|
|
|
|
# DEFINE THE CXX PATH
|
|
CXX=/usr/bin/g++
|
|
|
|
BUILD_DIR=build
|
|
DISTRIBUTE_DIR=distribute
|