采用cmake nvcc fatal : Unsupported gpu architecture 'compute_20

最新推荐文章于 2023-06-18 16:17:20 发布

fdd096030079

最新推荐文章于 2023-06-18 16:17:20 发布

阅读量1.7k

点赞数 2

本文链接：https://blog.csdn.net/fdd096030079/article/details/84451811

版权

环境：Ubuntu 18.04， caffe ssd版本

采用make来编译caffe的时候，报错：nvcc fatal : Unsupported gpu architecture 'compute_20

第一反应是Makefile.config文件中注释掉相关的行，比如https://blog.csdn.net/tjuyanming/article/details/79249964

但是我现在是用cmake来编译caffe的，没有用到Makefile.config文件。

网上看到这个方法：https://blog.csdn.net/renhanchi/article/details/80493232

依然没有用。

后来我自己去cmake的相关文件里查找，果然被我找到了。在这个文件里：caffe_ssd/cmake/Cuda.cmake

  1 if(CPU_ONLY)
  2   return()
  3 endif()
  4 
  5 # Known NVIDIA GPU achitectures Caffe can be compiled for.
  6 # This list will be used for CUDA_ARCH_NAME = All option
  7 #set(Caffe_known_gpu_archs "20 21(20) 30 35 50 52 61")
  8 set(Caffe_known_gpu_archs "30 35 50 52 61")
  9

这里把第7行修改为第八行就OK啦！