GXYM/DRRG

The solution for "Unsupported gpu architecture 'compute_86'"

Shualite opened this issue · 1 comments

When i used 3090 cuda11.0+ to run the code 'make -j4' in './csrc', i met error 'Unsupported gpu architecture 'compute_86''.

The way to solve this error by two solutions:

  1. change your cuda version below cuda11. (Maybe it will be limited by your GPU version).
  2. try to add nvcc params in setup.py:
    extra_compile_args["nvcc"] = [ "-DCUDA_HAS_FP16=1", "-D__CUDA_NO_HALF_OPERATORS__", "-D__CUDA_NO_HALF_CONVERSIONS__", "-D__CUDA_NO_HALF2_OPERATORS__", "-gencode=arch=compute_80,code=sm_80", ]