device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
我发现我没下载CUDA
于是先在NVIDIA上找到了对应的CUDA驱动,要10.0.132的driver
然后我就下载安装了,这个很容易,但是我发现运行完还是显示我没有CUDA,经查发现是我的torch版本太高了,而且我电脑没安装pytorch.
CUDA10.0.130-pytorch1.2.0-torch0.4.0这个版本是可以正常运行的
我是直接从百度网盘下载的,pytorch下载真的太慢了,而且镜像我没整明白,最后在anaconda中执行的命令行安装的。