Update version string

CMakeLists.txt

@@ -3,7 +3,7 @@ MESSAGE(STATUS "Using CMake ${CMAKE_VERSION}")
 
 SET(CUBEAILIB_VERSION_MAJOR 0)
 SET(CUBEAILIB_VERSION_MINOR 1)
-SET(CUBEAILIB_VERSION_PATCH 7)
+SET(CUBEAILIB_VERSION_PATCH 8)
 SET(CUBEAILIB_VERSION "${CUBEAILIB_VERSION_MAJOR}.${CUBEAILIB_VERSION_MINOR}.${CUBEAILIB_VERSION_PATCH}")
 
 
@@ -12,7 +12,7 @@ MESSAGE(STATUS "CubeAI lib Version ${CUBEAILIB_VERSION}")
 
 # if PyTorch has been compiled with CUDA support
 # then we need to enable this flag
-SET(USE_CUDA ON)
+SET(USE_CUDA OFF)
 
 IF(USE_CUDA)
 	PROJECT(cubeai VERSION ${CUBEAILIB_VERSION} LANGUAGES CXX C CUDA)
@@ -52,7 +52,7 @@ SET(CMAKE_LINKER_FLAGS "-pthread")
 
 SET(USE_OPENMP ON)
 SET(USE_LOG ON)
-SET(USE_PYTORCH OFF)
+SET(USE_PYTORCH ON)
 
 SET(USE_OPENCV OFF)
 SET(USE_RLENVS_CPP ON)
@@ -72,9 +72,8 @@ IF(USE_CUDA)
 	# we need to be careful here as PyTorch may not
 	# accept what we set here. PyTorch version ?
 	# needs at least C++17 to work
-	SET(CMAKE_CUDA_STANDARD 11)
+	SET(CMAKE_CUDA_STANDARD 17)
 	SET(CMAKE_CUDA_FLAGS ${CMAKE_CUDA_FLAGS}; "-O3 ${ARCH_FLAGS}")
-	#SET(CMAKE_CUDA_ARCHITECTURES 60)
 
 ENDIF()