-
Notifications
You must be signed in to change notification settings - Fork 1
/
CMakeLists.txt
27 lines (22 loc) · 1.17 KB
/
CMakeLists.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
set (PYTORCH_SRC_LIB_PATH ${PYTORCH_SRC_PATH}/build/lib)
################################### shared libs ########################################
set(shared_libs torch torch_cpu c10 torch_cuda c10_cuda rt pthread log4cxx)
link_directories(${TORCH_LIB_PATH})
link_directories(${CONDA_LIB_PATH})
link_directories(${CONDA_LIB64_PATH})
################################### tensorboard libs ####################################
set(tensorboard_libs ${TENSORBOARD_PATH}/build/libtensorboard_logger.a)
######################################### protobuf ######################################
set(proto_libs ${PYTORCH_SRC_LIB_PATH}/libprotobuf.a)
######################################## gym libs #######################################
set(PYTORCH_CPP_RL_LIB_PATH ${PYTORCH_CPP_RL_PATH}/example/lib)
set(GYM_SRC_BUILD_PATH ${GYM_SRC_PATH}/build)
set(GYM_SRC_EXAMPLE_BUILD_PATH ${GYM_SRC_BUILD_PATH}/example/lib)
set(pytorch_cpp_rl_lib ${GYM_SRC_BUILD_PATH}/libcpprl.a ${GYM_SRC_EXAMPLE_BUILD_PATH}/libzmq/lib/libzmq.a)
add_subdirectory(gymtest)
add_subdirectory(trials)
#add_subdirectory(lmdb)
add_subdirectory(envtest/)
add_subdirectory(testutils)
add_subdirectory(tensorboard)
add_subdirectory(a3c)