我正在尝试在与 GTX 1060 显卡一起使用的本地计算机上将一些 OpenMP 卸载到 GPU。我所有的 CUDA 和 Cublas 示例都运行良好。但是,当我尝试运行一些 OpenMP 卸载时,它根本不起作用。为了支持 OpenMP 5.0,我编译了 GCC 10.2.0 工具链。经过一番调试,我发现 OpenMP 运行时看不到任何设备。例如,此代码显示为零:
#include <omp.h>
#include <stdio.h>
int main() {
printf("%d\n", omp_get_num_devices());
return 0;
}
但是,Nvidia 工具链已启动并正在运行:
$ nvidia-smi
Sun Feb 21 23:06:40 2021
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.51.06 Driver Version: 450.51.06 CUDA Version: 11.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce GTX 106... Off | 00000000:1D:00.0 Off | N/A |
| 0% 37C P8 12W / 200W | 584MiB / 6075MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
那么我错过了什么?OpenMP 运行时如何找到设备?
编辑:
我正在附加有关我的编译器的信息:
$ gcc -v
Using built-in specs.
COLLECT_GCC=gcc
COLLECT_LTO_WRAPPER=/opt/gcc/10.2.0/libexec/gcc/x86_64-pc-linux-gnu/10.2.0/lto-wrapper
Target: x86_64-pc-linux-gnu
Configured with: ./configure --prefix=/opt/gcc/10.2.0/
Thread model: posix
Supported LTO compression algorithms: zlib
gcc version 10.2.0 (GCC)
该代码是使用以下命令编译的:
gcc -fopenmp simple.c