import torch print(torch.cuda.is_available())怎么改GPU

如果你的电脑有多个GPU，你可以使用以下代码来指定使用哪一个GPU：

import torch

device = torch.device("cuda:0")  # 指定使用 cuda 设备，编号为 0
print(torch.cuda.is_available())

# 在模型中指定使用 device
model.to(device)

# 在数据上指定使用 device
data = data.to(device)

其中，`cuda:0` 表示使用第一个GPU，如果有多个GPU，可以使用 `cuda:1`、`cuda:2` 等来指定使用其他的GPU。

如果你只有一个GPU，可以使用以下代码来设置 GPU：

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"  # 指定使用 cuda 设备，编号为 0
print(torch.cuda.is_available())

# 在模型中指定使用默认的 cuda 设备
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)

# 在数据上指定使用默认的 cuda 设备
data = data.to(device)

其中，`CUDA_VISIBLE_DEVICES` 环境变量可以用来指定使用哪一个GPU，编号为 0 表示使用第一个GPU。如果你的电脑有多个GPU，可以使用 `CUDA_VISIBLE_DEVICES=0,1` 来指定使用多个GPU。