Python model.to（torch.device（“cuda”）错误-未在启用cuda的情况下编译torch

我一直在尝试在VisualStudio代码中从这里运行代码

我得到model.to（torch.device（“cuda”）的一个错误

---

它还说AssertionError:Torch未在启用CUDA的情况下编译

我非常确定Cuda已为Torch启用（我有NVIDIA GeForce GTX 1650）

下面是代码的一部分

training_data_path = ".."
model_path = ".."
df = pd.read_csv("..")
device = "cuda"
epochs = 50
train_bs = 32
valid_bs = 16

df_train = df[df.kfold != fold].reset_index(drop=True)
df_valid = df[df.kfold == fold].reset_index(drop=True)

model = SEResNext50_32x4d(pretrained = "imagenet")
model.to(torch.device("cuda"))

我不明白为什么它不认识它。

---

请提供帮助。

“Torch未在启用CUDA的情况下编译”的意思与上面所说的完全一致——您安装的Pytorch版本尚未在CUDA支持下构建。不管你是否安装CUDA，都不会有什么不同。您需要找到并安装不同版本的Pytorchi。我使用官方网站上的命令“conda install pytorch torchvision torchaudio cudatoolkit=10.2-c pytorch”，该命令应该为这两个版本安装最新版本。我应该改变什么，或者做什么不同的事情来让它工作？