nms::setup_windows_cuda Namespace Reference

Classes
class	CUDA_build_ext
Variables
list	cuda_libs = ['cublas']
tuple	cudamat_ext
string	lib_dir = 'lib/x64'
tuple	numpy_include = np.get_include()
string	nvcc_bin = 'nvcc.exe'
list	nvcc_compile_args = ['-O', '--ptxas-options=-v', '-arch=compute_35', '-code=sm_35,sm_52,sm_61', '-c', '--compiler-options=-fPIC']
tuple	PATH = os.environ.get('PATH')

---

Variable Documentation

list nms::setup_windows_cuda::cuda_libs = ['cublas']

Definition at line 16 of file setup_windows_cuda.py.

tuple nms::setup_windows_cuda::cudamat_ext

Initial value:

Extension('gpu_nms',
                        sources=[
                                'gpu_nms.cu'
                                ],
                        language='c++',
                        libraries=cuda_libs,
                        extra_compile_args=nvcc_compile_args,
                        include_dirs = [numpy_include, 'C:\\Programming\\CUDA\\v8.0\\include'])

Definition at line 31 of file setup_windows_cuda.py.

string nms::setup_windows_cuda::lib_dir = 'lib/x64'

Definition at line 18 of file setup_windows_cuda.py.

tuple nms::setup_windows_cuda::numpy_include = np.get_include()

Definition at line 26 of file setup_windows_cuda.py.

string nms::setup_windows_cuda::nvcc_bin = 'nvcc.exe'

Definition at line 17 of file setup_windows_cuda.py.

tuple nms::setup_windows_cuda::nvcc_compile_args = ['-O', '--ptxas-options=-v', '-arch=compute_35', '-code=sm_35,sm_52,sm_61', '-c', '--compiler-options=-fPIC']

Definition at line 14 of file setup_windows_cuda.py.

tuple nms::setup_windows_cuda::PATH = os.environ.get('PATH')

Definition at line 6 of file setup_windows_cuda.py.