pytorch cpu版怎么换成gpu版本

要将PyTorch CPU版本切换为GPU版本，需要执行以下步骤：

1. 确认你的电脑上有可用的GPU，并且已经安装了NVIDIA CUDA驱动和cuDNN库。

2. 安装PyTorch GPU版本。可以通过以下命令来安装：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

其中，cu111表示你的CUDA版本，如果你的CUDA版本不是11.1，则需要相应地更改。

3. 在代码中，将所有的Tensor转换为CUDA Tensor。可以使用以下代码来将Tensor转换为CUDA Tensor：

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

其中，x是你的Tensor。

这样，你就可以将PyTorch CPU版本切换为GPU版本了。