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


尴尬的是,官网好像比较支持CUDA9.2CUDA10.1,没有CUDA10.0的一席之地…
在这里插入图片描述

从官网中找到稳定版下载的网址
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就可以了

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

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

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

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

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

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐