怎么用gpu跑python程序

首先，需要确保你的电脑或服务器上已经安装了GPU驱动和CUDA工具包。然后，你需要安装TensorFlow或PyTorch等支持GPU加速的深度学习框架。

在编写Python程序时，你需要在代码中添加以下几行代码以启用GPU加速：

对于TensorFlow：

import tensorflow as tf
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

对于PyTorch：

import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

接下来，你可以使用以下命令运行Python程序：

CUDA_VISIBLE_DEVICES=0 python your_program.py

其中，CUDA_VISIBLE_DEVICES用于指定使用哪个GPU设备，0表示使用第一块GPU设备。你可以根据实际情况修改设备编号。