cuda10.0环境安装Pytorch

前置环境需要安装cuda10.0和对应的cudnn
不在赘述,如需查看请参考这两篇博客(面向小白的,十分详细):
手把手教你win10安装Paddlepaddle-GPU(一)
手把手教你win10安装Paddlepaddle-GPU(二)


尴尬的是,官网好像比较支持CUDA9.2CUDA10.1,没有CUDA10.0的一席之地…
cuda10.0环境安装Pytorch_第1张图片

从官网中找到稳定版下载的网址
https://download.pytorch.org/whl/torch_stable.html

搜索cu100,可以发现,Pytorch对windows的cuda10.0可真是太不友好了
最高版本也只是支持到Pytorch1.1.0,真的尴尬

通过查看Pytorchtorchvision的对应关系:
https://blog.csdn.net/qq_40263477/article/details/106577790

我们下载torchvision 0.3.0就可以了

这两个下载链接分别是:
pytorch:https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl

torchvision:https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

扔到某雷里边下载,速度还行

如果链接变了,大家从那个稳定版的下载链接中找一下就行
最后验证一下:

import torch
torch.cuda.is_available() # True 就完事儿了

你可能感兴趣的:(cuda,gpu,windows,pytorch,python)