Windows10下GPU版本的pytorch1.6.0+torchvision0.7.0安装教程(支持CUDA10.2,含whl快速下载地址)

本文介绍了在Windows10上安装GPU版本PyTorch1.6.0和torchvision0.7.0的详细步骤,包括CUDA10.2和cuDNN的配合使用,以及如何验证PyTorch是否成功利用GPU加速。
摘要由CSDN通过智能技术生成

本文实现了Windows10下GPU版本的pytorch的安装,用到的软件主要包括:CUDA 10.2 + cuDNN + Anaconda + pytorch1.6.0 + torchvision0.7.0。这里说明一下,大家可以根据自己安装的CUDA选择适合自己的pytorch和torchvision版本进行下载和安装。

CUDA和cuDNN的安装可以参考:Windows10下安装tensorflow-gpu(2.2.0)安装教程(避坑+保姆式教学)

一、whl下载

如果直接使用anaconda安装,即使添加了清华源速度也很慢,所以我们下载whl进行安装会大大减少时间成本。

pytorch的各种whl的下载地址:https://download.pytorch.org/whl/torch_stable.html

因为我的CUDA是10.2版本,所以在cu102的目录下找到了自己需要的whl,分别是

  • cu102/torch-1.6.0-cp36-cp36m-win_amd64.whl
  • cu102/torchvision-0.7.0-cp36-cp36m-win_amd64.whl

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值