gpt4 book ai didi

cuda - 我可以调用什么实用程序/二进制文件来确定 nVIDIA GPU 计算能力?

转载 作者:行者123 更新时间:2023-12-04 15:19:31 25 4
gpt4 key购买 nike

假设我有一个安装了单个 GPU 的系统,并假设我还安装了最新版本的 CUDA。

我想确定我的 GPU 的计算能力是什么。如果我可以编译代码,那就很容易了:

#include <stdio.h>
int main() {
cudaDeviceProp prop;
cudaGetDeviceProperties(&prop, 0);
printf("%d", prop.major * 10 + prop.minor);
}

但是 - 假设我想在不编译的情况下做到这一点。我可以吗?我以为 nvidia-smi可能对我有帮助,因为它可以让您查询有关设备的各种信息,但它似乎不允许您获得计算能力。也许我还能做些什么?也许可以通过 /proc 看到一些东西或系统日志?

编辑:这旨在在构建之前在我无法控制的系统上运行。因此它必须具有最小的依赖关系,在命令行上运行并且不需要 root 权限。

最佳答案

不幸的是,目前的答案似乎是“否”,并且需要编译程序或使用在其他地方编译的二进制文件。
编辑:我已经针对这个问题调整了一个解决方法 - 一个独立的 bash script它编译一个小的内置 C 程序来确定计算能力。 (使用 CMake 调用特别有用,但可以独立运行。)
另外,我已经提交了 feature-requesting bug reportnVIDIA对这个。
这是脚本,假设版本为 nvcc在你的路上:

//usr/bin/env nvcc --run "$0" ${1:+--run-args "${@:1}"} ; exit $?
#include <cstdio>
#include <cstdlib>
#include <cuda_runtime_api.h>

int main(int argc, char *argv[])
{
cudaDeviceProp prop;
cudaError_t status;
int device_count;
int device_index = 0;
if (argc > 1) {
device_index = atoi(argv[1]);
}

status = cudaGetDeviceCount(&device_count);
if (status != cudaSuccess) {
fprintf(stderr,"cudaGetDeviceCount() failed: %s\n", cudaGetErrorString(status));
return -1;
}
if (device_index >= device_count) {
fprintf(stderr, "Specified device index %d exceeds the maximum (the device count on this system is %d)\n", device_index, device_count);
return -1;
}
status = cudaGetDeviceProperties(&prop, device_index);
if (status != cudaSuccess) {
fprintf(stderr,"cudaGetDeviceProperties() for device device_index failed: %s\n", cudaGetErrorString(status));
return -1;
}
int v = prop.major * 10 + prop.minor;
printf("%d\n", v);
}

关于cuda - 我可以调用什么实用程序/二进制文件来确定 nVIDIA GPU 计算能力?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40695455/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com