最近在安装一个光流库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

完成安装

更多推荐