本文实现了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


二、安装pytorch1.6.0 + torchvision0.7.0
打开Anaconda Prompt,切换到whl的路径,依次输入以下命令即可进行安装即可。因为pytorch需要先安装依赖包future,torchvision需要先安装依赖包Pillow,用清华源提前安装可以节约时间。
Anaconda相关操作可以参考Anaconda虚拟环境安装Jupyter Notebook并更换默认工作目录详细教程及使用攻略
# 激活

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

5550

被折叠的 条评论
为什么被折叠?



