查看问题
我们可以使用上面的代码，查看 torch 的版本。
以及是否使用了 cuda，可以看到 torch 的版本后面有个 +cpu, 而 cuda 是否可用返回的是 none 说明没有用到 cuda。
这时你可能会疑惑：我明明用的的英伟达显卡，有 cuda 啊。
其实是 torch 版本的问题。
卸载与重装 torch
先卸载已经安装的 torch：
pip uninstall torch torchvision torchaudio然后安装新的 torch：
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121这里安装的版本是 cuda 12.1
安装可能需要比较长的时间
查看支持的 cuda 版本
打开 cmd。
输入：nvidia-smi
这里的 cuda version 就是支持的最高版本的 cuda，可以安装比它版本低的。
查看是否解决问题
再次运行。
可以看到返回的结果是 True 了。