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)
还没有任何评论哟~
