sm61

Author: zhangshiyang

README.md

@@ -1,5 +1,57 @@
 # CV-CUDA
 
+CV-CUDA(0.5.0) for Pascal(sm6.1) and lower version gcc
+
+## build
+  
+```shell  
+git clone [email protected]:torchpipe/CV-CUDA.git
+git submodule update --init
+
+img_name=nvcr.io/nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04
+
+docker run  --name=cvcuda_fork   --network=host --gpus=all  --ulimit memlock=-1 --ulimit stack=67108864  --privileged=true  -v `pwd`:/workspace  -w/workspace  -it $img_name  /bin/bash 
+```
+
+  and then
+
+```
+source build_env.sh
+./ci/build.sh
+
+ls -alh build-rel/lib/
+
+```
+
+(optinal) pack
+```
+cd build-rel
+cpack . -G TXZ
+
+ls -alh nvcv-*
+    
+```
+
+
+## direct download
+
+see https://github.com/torchpipe/CV-CUDA/releases
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+original readme
+--------------------------------------------------------------------------
 [![License](https://img.shields.io/badge/License-Apache_2.0-yellogreen.svg)](https://opensource.org/licenses/Apache-2.0)
 
 ![Version](https://img.shields.io/badge/Version-v0.5.0--beta-blue)

build_env.sh

@@ -0,0 +1,9 @@
+apt-get update -y
+  apt-get install -y  ninja-build  libssl-dev wget
+ 
+wget https://cmake.org/files/v3.20/cmake-3.20.1-linux-x86_64.tar.gz -O /tmp/cmake-3.20.1-linux-x86_64.tar.gz
+ cd /tmp
+ tar zxvf cmake-3.20.1-linux-x86_64.tar.gz
+mv cmake-3.20.1-linux-x86_64 /opt/cmake-3.20.1
+ln -sf /opt/cmake-3.20.1/bin/*  /usr/bin/
+cd -

ci/build.sh

@@ -67,7 +67,8 @@ mkdir -p "$build_dir"
 # Set build configuration depending on build type ====================
 
 # Common config
-cmake_args="-DBUILD_TESTS=1"
+cmake_args="-DBUILD_TESTS=0"
+ENABLE_PYTHON=0
 
 if [[ "$ENABLE_PYTHON" = '0' || "$ENABLE_PYTHON" = 'no' ]]; then
     cmake_args="$cmake_args -DBUILD_PYTHON=0"
@@ -94,9 +95,9 @@ esac
 
 # Configure build toolchain ===========================================
 
-# Make sure we use most recent gcc-11.x
-CC=${CC:=$(find /usr/bin/gcc-11* | sort -rV | head -n 1)}
-CXX=${CXX:=$(find /usr/bin/g++-11* | sort -rV | head -n 1)}
+# Make sure we use most recent gcc-9.x
+CC=${CC:=$(find /usr/bin/gcc-9* | sort -rV | head -n 1)}
+CXX=${CXX:=$(find /usr/bin/g++-9* | sort -rV | head -n 1)}
 
 cmake_args="${cmake_args} -DCMAKE_C_COMPILER=$CC -DCMAKE_CXX_COMPILER=$CXX"
 
@@ -130,7 +131,7 @@ done
 
 # Create build tree
 cmake -B "$build_dir" "$source_dir"  \
-    -DBUILD_TESTS=1 \
+    -DBUILD_TESTS=0 \
     $cmake_args \
     $user_args
 

cmake/ConfigCUDA.cmake

@@ -21,10 +21,10 @@ list(GET CUDA_VERSION_LIST 2 CUDA_VERSION_PATCH)
 find_package(CUDAToolkit ${CUDA_VERSION_MAJOR}.${CUDA_VERSION_MINOR} REQUIRED)
 
 # CUDA version requirement:
-# - to use gcc-11 (11.7)
+# - to use gcc-11 (11.6)
 
-if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS "11.7")
-    message(FATAL_ERROR "Minimum CUDA version supported is 11.7")
+if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS "11.6")
+    message(FATAL_ERROR "Minimum CUDA version supported is 11.6")
 endif()
 
 set(CMAKE_CUDA_STANDARD ${CMAKE_CXX_STANDARD})
@@ -43,6 +43,7 @@ if(NOT USE_CMAKE_CUDA_ARCHITECTURES)
     else()
         # All architectures we build sass for
         list(APPEND CMAKE_CUDA_ARCHITECTURES
+            61-real # Pascal - gp10x/GeForce
              70-real # Volta  - gv100/Tesla
              75-real # Turing - tu10x/GeForce
              80-real # Ampere - ga100/Tesla