解决错误nvcc fatal : Unsupported gpu architecture ‘compute_86‘
最近在安装一个光流库spatial_coorelation_sampler==0.5的时候,出现一个报错
conda环境pip安装失败,于是github下载源码安装。出现错误如下
nvcc fatal : Unsupported gpu architecture 'compute_86'
这显示我的显卡不支持compute_86架构,于是我查了下,我的显卡支持的架构,我是A100,
在如下链接查看发现只支持compute_80
所以在安装光流库之前,得这是安装环境的GPU架构
export TORCH_CUDA_ARCH_LIST="8.0"
然后再执行安装命令
python set_up.py install
完成安装