Unsupported gpu architecture 'compute_86' on Windows 10 RTX 3070

[stelmath]

OS: Windows 10, 64 bit

 λ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Wed_Jul_22_19:09:35_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.0, V11.0.221
Build cuda_11.0_bu.relgpu_drvr445TC445_37.28845127_0

CUDA Toolkit 11.0 Update 1
cuDNN v8.0.4 (September 28th, 2020), for CUDA 11.0
torch==1.7.1+cu110
GPU: RTX 3070 with Driver Version: 457.09 (the drivers of the CUDA Toolkit [451.82] do NOT support RTX 3070)
Also:

deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 11.1, CUDA Runtime Version = 11.0, NumDevs = 1
Result = PASS

Hello, when using: pip install -v --no-cache-dir . the installation works fine and I am able to use apex.
However, when using: pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .
I get:
nvcc fatal : Unsupported gpu architecture 'compute_86'

I removed check_cuda_torch_binary_vs_bare_metal() like this suggests but, I still get the same error. Any help?

[untitledvirtualperson]

Echoing this issue with a 3090/Win 10 and Conda env. Currently on CUDA 11.0 and will try 11.1, though expecting the same issue.

[y-rokutan]

Hi @stelmath
I've just encountered the same problem, and typing export TORCH_CUDA_ARCH_LIST="7.5" fixed it. Can you try this and tell us the result?

[stelmath]

@y-rokutan yeah I tried that but did not work. Nevermind, I found some forks of apex, plus some hacks that solve this. Thanks.

[lyuweiwang]

@y-rokutan yeah I tried that but did not work. Nevermind, I found some forks of apex, plus some hacks that solve this. Thanks.

Could you please post the fork's name or the website? Thanks

[stelmath]

@lyuweiwang sorry I can't really find it now. Better switch to a Linux OS so that apex works properly

[javierrodenas]

Hello @stelmath,

As mentioned on another comment, export TORCH_CUDA_ARCH_LIST="7.5" worked for me (linux).

Best