运行在 GPU 上轻松编写 C++ 代码 lik Python 语言

Run C++ code easily on GPU lik Python language

c++
gpu

为了运行 Python 使用 GPU 编码（为此安装所有库之后），可以在代码前添加这些行：

device_name= tf.test.gpu_device_name()
if device_name!='/device:GPU:0':
    raise SystemEror('GPU device not found')
else:
    print('GPU found!')

with tf.device("/device:GPU:0"):
    #python code here...

是否有与运行 GPU 上现有 C++ 代码类似的方法？我搜索了一下，发现应该安装其他库并重新编写 C++ 代码。我更喜欢类似于 Python.

的东西（如果存在的话）

有什么想法吗？

C++ 没有任何内置的 GPU 编程工具，您需要使用 OpenCL（或 CUDA，如果您有 Nvidia GPU）之类的东西进行计算，或者 OpenGL/Vulkan 或其他图形库用于图形。