search

sczhou / CodeFormer

[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer
Other
15.93k stars 3.36k forks source link

RTX 4090 Error: nvrtc: error: invalid value for --gpu-architecture (-arch) #167

Open Deadstarrr opened 1 year ago

Deadstarrr commented 1 year ago

I'm getting this error when I try to use codeformer. Does it not work with a 4090?

nvrtc: error: invalid value for --gpu-architecture (-arch)

Anyone know how to fix this?

huangxin168 commented 1 year ago

maybe you need to upgrade your cuda-toolkit to 11.8 or newer

Sun-ZhenXing commented 3 months ago

PyToch 1.10 & CUDA 11.3 don't support RTX 4090, update CUDA >= 11.8:

pip3 install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118