Fix some bugs on Ubuntu.

CMakeLists.txt

@@ -159,7 +159,13 @@ if(USE_CUDA)
         set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "${CUDA_LIB_DIR}libnppc_static.a")
         set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "${CUDA_LIB_DIR}libnppc_static.a")
         set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "${CUDA_LIB_DIR}libcudadevrt.a")
         set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "${CUDA_LIB_DIR}libcudadevrt.a")
         set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "${CUDA_LIB_DIR}libcurand_static.a")
         set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "${CUDA_LIB_DIR}libcurand_static.a")
-        set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "/usr/lib64/libdl.so.2")
+        if(EXISTS "/usr/lib64/libdl.so.2")
+            set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "/usr/lib64/libdl.so.2")
+        elseif(EXISTS "/lib/x86_64-linux-gnu/libdl.so.2")
+            set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "/lib/x86_64-linux-gnu/libdl.so.2")
+        elseif(EXISTS "/lib64/libdl.so.2")
+            set(CUDA_LIB_PATH ${CUDA_LIB_PATH} "/lib64/libdl.so.2")
+        endif()
     endif()
     endif()
 endif()
 endif()
 
 
@@ -294,4 +300,4 @@ else()
         target_link_libraries(${NIUTENSOR_EXE} ${ALL_LIB} ${FLAG})
         target_link_libraries(${NIUTENSOR_EXE} ${ALL_LIB} ${FLAG})
     endif()
     endif()
     message(STATUS "${MESS}")
     message(STATUS "${MESS}")
-endif()
+endif()
\ No newline at end of file