怎么用gpu跑python程序

首先，你需要安装好CUDA和cuDNN，然后你可以在Python中使用GPU来运行程序，具体步骤如下：

1. 安装GPU版本的TensorFlow或PyTorch等深度学习框架。

2. 在你的Python程序中，使用以下代码来检查是否有可用的GPU：

   import tensorflow as tf
   print(tf.test.is_gpu_available())

如果返回True，则说明你的GPU可以用来运行程序。

3. 在你的Python程序中，使用以下代码来指定使用哪个GPU：

   import os
   os.environ["CUDA_VISIBLE_DEVICES"] = "0"  # 使用第0个GPU

如果你有多个GPU，可以指定使用哪个GPU来运行程序。

4. 在你的Python程序中，使用以下代码来指定使用GPU来运行程序：

   import tensorflow as tf
   with tf.device('/GPU:0'):  # 使用第0个GPU
       # 在这里编写你的代码

如果你使用的是PyTorch，可以使用以下代码来指定使用GPU来运行程序：

   import torch
   device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")  # 使用第0个GPU
   model.to(device)

这样就可以使用GPU来加速你的程序了。