autonomousvision / occupancy_networks

This repository contains the code for the paper "Occupancy Networks - Learning 3D Reconstruction in Function Space"
https://avg.is.tuebingen.mpg.de/publications/occupancy-networks
MIT License
1.5k stars 292 forks source link

Unsupported gpu architecture 'compute_75' during installation #26

Open sunglyoungKim opened 4 years ago

sunglyoungKim commented 4 years ago

python3 setup.py build_ext --inplace running build_ext building 'im2mesh.dmc.ops._cuda_ext' extension gcc -pthread -B /home/sunglyoung_119/miniconda3/envs/mesh_funcspace/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -I/home/sunglyoung_119/.local/lib/python3.6/site-packages/torch/include -I/home/sunglyoung_119/.local/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -I/home/sunglyoung_119/.local/lib/python3.6/site-packages/torch/include/TH -I/home/sunglyoung_119/.local/lib/python3.6/site-packages/torch/include/THC -I/home/sunglyoung_119/miniconda3/envs/mesh_funcspace/include/python3.6m -c im2mesh/dmc/ops/src/extension.cpp -o build/temp.linux-x86_64-3.6/im2mesh/dmc/ops/src/extension.o -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=_cuda_ext -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++11 cc1plus: warning: command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++ /usr/bin/nvcc -I/home/sunglyoung_119/.local/lib/python3.6/site-packages/torch/include -I/home/sunglyoung_119/.local/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -I/home/sunglyoung_119/.local/lib/python3.6/site-packages/torch/include/TH -I/home/sunglyoung_119/.local/lib/python3.6/site-packages/torch/include/THC -I/home/sunglyoung_119/miniconda3/envs/mesh_funcspace/include/python3.6m -c im2mesh/dmc/ops/src/curvature_constraint_kernel.cu -o build/temp.linux-x86_64-3.6/im2mesh/dmc/ops/src/curvature_constraint_kernel.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options '-fPIC' -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=_cuda_ext -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

I am using RTX2080 so I don't think the Cuda 9.2 architecture recognize the GPU. However, is it anyway I can add 'compute_75' as a support GPU? If I can do, or may I know what file should I look into?