cuda10.0环境安装Pytorch

前置环境需要安装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 就完事儿了
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值