tiny-cuda-nn / detect_tcnn_cuda_architectures.cu
camenduru's picture
thanks to nvlabs ❤
9e0a83e
raw
history blame contribute delete
409 Bytes
#include <cuda_runtime.h>
#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);
if (device < count - 1) std::printf(";");
}
}
return 0;
}