search

submission2019 / cnn-quantization

Quantization of Convolutional Neural networks.
237 stars 59 forks source link

error: command '/usr/bin/nvcc' failed with exit status 1 #12

Open georgetseng1128 opened 4 years ago

georgetseng1128 commented 4 years ago

Hi, I build cuda kernels for GEMMLOWP is "./build_all.sh" step have error:

**************************************************************
Building int quantization kernels
**************************************************************
running install
running bdist_egg
running egg_info
writing int_quantization.egg-info/PKG-INFO
writing dependency_links to int_quantization.egg-info/dependency_links.txt
writing top-level names to int_quantization.egg-info/top_level.txt
reading manifest file 'int_quantization.egg-info/SOURCES.txt'
writing manifest file 'int_quantization.egg-info/SOURCES.txt'
installing library code to build/bdist.linux-x86_64/egg
running install_lib
running build_ext
building 'int_quantization' extension
creating build
creating build/temp.linux-x86_64-3.6
x86_64-linux-gnu-gcc -pthread -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/george/work/cnn-quantization/venv3/lib/python3.6/site-packages/torch/include -I/home/george/work/cnn-quantization/venv3/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -I/home/george/work/cnn-quantization/venv3/lib/python3.6/site-packages/torch/include/TH -I/home/george/work/cnn-quantization/venv3/lib/python3.6/site-packages/torch/include/THC -I/usr/include/python3.6m -I/home/george/work/cnn-quantization/venv3/include/python3.6m -c int_quantization.cpp -o build/temp.linux-x86_64-3.6/int_quantization.o -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=int_quantization -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++11
/usr/bin/nvcc -I/home/george/work/cnn-quantization/venv3/lib/python3.6/site-packages/torch/include -I/home/george/work/cnn-quantization/venv3/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -I/home/george/work/cnn-quantization/venv3/lib/python3.6/site-packages/torch/include/TH -I/home/george/work/cnn-quantization/venv3/lib/python3.6/site-packages/torch/include/THC -I/usr/include/python3.6m -I/home/george/work/cnn-quantization/venv3/include/python3.6m -c gemmlowp.cu -o build/temp.linux-x86_64-3.6/gemmlowp.o -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options '-fPIC' -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=int_quantization -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=sm_75 -std=c++11
nvcc fatal   : Unsupported gpu architecture 'compute_75'
error: command '/usr/bin/nvcc' failed with exit status 1
Done
**************************************************************

My GPU is RTX2080Ti CUDA Version: 10.1 Ubuntu 18.04

and command 'nvcc -V' have return information:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

Please help.