nvcc fatal : Unsupported gpu architecture ‘compute_86‘

GPU： RTX3090
Linux： Ubuntu 16.04
pytorch: 1.7.0
python: 3.7.9
cuda version: V11.0.194
GCC version: 5.2.0

安装apex的过程中，按照官网的教程真是一波三折~

官网的教程如下：

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

但是在执行最后一行脚本时，遇到很多问题。
首先，安装时要保证cuda的版本与pytorch中对应的cuda版本一致。
一般可以通过在.bashrc文件中添加如下脚本指定cuda版本（服务器里有多个cuda版本时）。

当cuda版本没问题时，此时安装过程中又遇到如下问题：

主要原因应该是硬件能够支持的算力比较高，能达到8.6，但是cuda11.0支持不了这么高的算力，通过下述脚本，设置环境变量，降低算力要求，即可：

export TORCH_CUDA_ARCH_LIST="7.5"

直接在终端输入上述脚本后，再执行安装命令，即可发现安装过程如下：