detect_cuda_archs.cu

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

猜你喜欢

转载自www.cnblogs.com/cjyp/p/12023515.html