在复现pix2pix时遇到了问题，原因是cuda版本没装对。

1.先查cuda是否可用，输入代码：

import torch
print(torch.cuda.is_available())

若返回false，不可用。

2.再查自己电脑需要的cuda版本

终端输入代码：

   nvidia-smi

返回自己电脑的cuda，为cuda12.1

3.设置清华镜像源，并重新下载cuda12.1

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html

此处的下载版本得多查查，我在官网下载页面只给了我代码

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

不知道为啥下载不进去，每次都显示already 安装。所以最后在网上直接找到了上面的下载命令

4.再次print(torch.cuda.is_available())，发现cuda可用