admin 管理员组

文章数量: 1184232

本文实现了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下载
        • 二、安装pytorch1.6.0 + torchvision0.7.0
        • 三、验证pytorch能否使用gpu加速
        • 结尾 & 彩蛋

一、whl下载

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

pytorch的各种whl的下载地址：https://download.pytorch/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

二、安装pytorch1.6.0 + t

本文标签： 下载地址 版本 快速 教程 GPU

版权声明：本文标题：Windows10下GPU版本的pytorch1.6.0+torchvision0.7.0安装教程(支持CUDA10.2，含whl快速下载地址) 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1754522332a3011874.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。