码迷,mamicode.com
首页 > 其他好文 > 详细

detect_cuda_archs.cu

时间:2019-12-11 17:11:08      阅读:72      评论:0      收藏:0      [点我收藏+]

标签:ice   count   print   poll   arc   dock   das   ==   ade   

apollo@in_dev_docker:/apollo/ttt/caffe/build$ cat detect_cuda_archs.cu 
#include <cstdio>
int main()
{
  int count = 0;
  if (cudaSuccess != cudaGetDeviceCount(&count)) return -1;
  if (count == 0) return -1;
  for (int device = 0; device < count; ++device)
  {
    cudaDeviceProp prop;
    if (cudaSuccess == cudaGetDeviceProperties(&prop, device))
      std::printf("%d.%d ", prop.major, prop.minor);
  }
  return 0;
}
apollo@in_dev_docker:/apollo/ttt/caffe/build$ ./a.out 
5.0

detect_cuda_archs.cu

标签:ice   count   print   poll   arc   dock   das   ==   ade   

原文地址:https://www.cnblogs.com/cjyp/p/12023515.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!