Advertisement

Windows 10 系统下 GPU 版本的 PyTorch 1.6.0 + torchvision 0.7.0 安装指南(支持 CUDA 10.2,并附有.whl 快速下载链接)

阅读量:

本文实现了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文件
  • 安装PyTorch 1.6.0及其相关库
  • 验证PyTorch是否能够利用GPU进行加速
  • 结尾及彩蛋
一、whl下载

通常情况下,在使用Anaconda进行安装时(如果直接使用anaconda安装),即便加入清华大学的源库(即使添加了清华源),速度依然较慢;因此下载Wheel版本(whl)进行安装将节省大量时间(大大减少时间成本)。

PyTorch各个版本的Whl下载链接:https://download.pytorch.org/whl/torch_stable.html

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

  • cu102/torch-1.6

全部评论 (0)

还没有任何评论哟~