Python GPU 进程创建 - 使用 multiprocessing 和 torch

要在 Python 中创建一个占用 GPU 的进程，你可以使用 'multiprocessing' 模块。首先，你需要确保你的系统上已经安装了 'torch' 库，它是一个用于深度学习的 Python 库，可以使用 GPU 进行计算。

下面是一个示例代码，它创建了一个使用 GPU 的进程：

import torch
import multiprocessing as mp

def gpu_process():
    # 设置 GPU 设备
    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
    print('Running on:', device)

    # 进行一些 GPU 计算
    a = torch.tensor([1.0, 2.0, 3.0], device=device)
    b = torch.tensor([4.0, 5.0, 6.0], device=device)
    c = a + b

    # 打印结果
    print('Result:', c)

if __name__ == '__main__':
    # 创建进程并启动
    process = mp.Process(target=gpu_process)
    process.start()
    process.join()

在这个示例中，'gpu_process' 函数会检查当前系统是否有可用的 GPU 设备，然后在 GPU 上进行一些计算。你可以根据需要在 'gpu_process' 函数中添加更多的 GPU 计算。

注意，这个示例中使用了 'torch.tensor' 创建了一些张量，并在 GPU 上进行计算。如果你想在 GPU 上运行其他类型的计算，请根据需要进行修改。