nvmlinit.c

#include <assert.h>
#include <dlfcn.h>
#include <stdio.h>
#include <stdlib.h>

int main() {
    void *handle;
    int (*nvmlInit)(void);
    char *error;

    handle = dlopen("libnvidia-ml.so", RTLD_LAZY);
    if (!handle) {
        fprintf(stderr, "%s\n", dlerror());
        exit(EXIT_FAILURE);
    }
    dlerror();
    nvmlInit = (int (*)(void)) dlsym(handle, "nvmlInit");
    error = dlerror();
    if (error != NULL) {
        fprintf(stderr, "%s\n", error);
        exit(EXIT_FAILURE);
    }
    assert(nvmlInit() == 0);
    dlclose(handle);
    return EXIT_SUCCESS;
}