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版本没问题时ÿ