CUDA10.0.130适用的pytorch和torchversion版本

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

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

你可能感兴趣的:(pytorch,深度学习,人工智能)