栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

【异常问题】nvcc fatal : Unsupported gpu architecture ‘compute

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【异常问题】nvcc fatal : Unsupported gpu architecture ‘compute

目录
        • 异常问题:nvcc fatal : Unsupported gpu architecture 'compute_86' 解决方案
          • (1)报错信息
          • (2)系统环境
          • (3)原因分析
          • (4)解决方案

异常问题:nvcc fatal : Unsupported gpu architecture ‘compute_86’ 解决方案 (1)报错信息
/usr/local/cuda/bin/nvcc -DWITH_CUDA -I/home/shanshan/anaconda3/envs/SOLOv2/lib/python3.7/site-packages/torch/include -
I/home/shanshan/anaconda3/envs/SOLOv2/lib/python3.7/site-packages/torch/include/torch/csrc/api/include -
I/home/shanshan/anaconda3/envs/SOLOv2/lib/python3.7/site-packages/torch/include/TH -
I/home/shanshan/anaconda3/envs/SOLOv2/lib/python3.7/site-packages/torch/include/THC -I/usr/local/cuda/include -
I/home/shanshan/anaconda3/envs/SOLOv2/include/python3.7m -c mmdet/ops/nms/src/nms_kernel.cu -o build/temp.linux-x86_64-
3.7/mmdet/ops/nms/src/nms_kernel.o -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -
D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options '-fPIC' -D__CUDA_NO_HALF_OPERATORS__ -
D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H -
DTORCH_EXTENSION_NAME=nms_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_86,code=sm_86 -std=c++14

nvcc fatal   : Unsupported gpu architecture 'compute_86'
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1
(2)系统环境
  • GPU: RTX 3080ti
  • Linux: Ubuntu 18.04
  • Pytorch: 1.7.0
  • Python: 3.7.9
  • CUDA version: V11.0.194
  • GCC version: 5.2.0
(3)原因分析

应该是硬件能够支持的算力比较高,能达到8.6,但是cuda11.0支持不了这么高的算力。

(4)解决方案
  • 设置环境变量,降低算力要求
# 清除TORCH_CUDA_ARCH_LIST
unset TORCH_CUDA_ARCH_LIST
# 设置cuda算力为8.0
export TORCH_CUDA_ARCH_LIST="8.0"
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/915948.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号