layers installation "nvcc fatal : Unsupported gpu architecture 'compute_86'"

[thomasweng15]

When running python setup.py install for layers, I get the following error:

nvcc fatal : Unsupported gpu architecture 'compute_86'

I am running with a 3090 GPU and CUDA 11.1. Is there something I can change in the layers directory to fix this issue? Thanks for your help.

[liruiw]

I haven't seen this issue before but it looks like a CUDA version problem.

[thomasweng15]

I was able to resolve this by using CUDA 11.1 and specifying it as an env variable: CUDA_HOME=path/to/cuda python setup.py install.