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