在RTX4090上运行深度学习代码报错:ValueError: Unknown CUDA arch (8.9) or GPU not suppor。
原因是cuda(我这里是cuda11.0,最高支持8.6)的版本不支持当前算力(8.9)
解决办法,改算力:
sudo vim ~/.bashrc
# 在配置文件中添加如下一行
export TORCH_CUDA_ARCH_LIST=8.6 # 因为是CUDA11.0,对应的算力为8.6
source ~/.bashrc
nvidia给的显卡算力查询:CUDA GPU | NVIDIA 开发者