pytorch win10 安装 含辛茹苦教程

小编一直在搭建pytorch(gpu) 深度学习的开发环境,在对目标检测的学习上能够更上一层楼。已经安装好tensorflow和pytroch,tensorflow的安装较快,没啥需要进行说明,网上大把的教程可供参考。关于win10下 cuda和cudnn的安装方式这里不做说明,网上安装案例很多,可供参考。

pytorch 安装为何如此艰难

    按照网上大把教程所说,参考pytorch官网的Get Started,会给出对应系统,python和cuda版本给出对应的安装方式,可以使用conda和pip方式的下载和安装。

 conda是anaconda的安装方式。anaconda真是python开发工具的神器,强烈推荐读者使用。

     conda 安装pytorch时一直出现conda httperror和sslerror问题,查询多日无果,只好采取离线安装的方式。

离线安装

    从百度云中下载对应版本的的安装包,我的是 pytorch-0.3.1-py35_cuda90_cudnn7he774522_2.tar.bz2。

百度云如下:(里面还有很多的,不过没有pytorch-0.4的)

      https://pan.baidu.com/s/19-gsCksVpNpMWWigmNMHvQ

     密码:zqa1

    下载完成后 ,将其放到对应的目录下: C:\Users\admin>

    采用对应的命令进行安装conda install pytorch-0.3.1-py35_cuda90_cudnn7he774522_2.tar.bz2,很快便呢个完成安装

         Downloading and Extracting Packages
        * -None    |     | ############################################################################## | 100%
        Preparing transaction: done
        Verifying transaction: done
        Executing transaction: done

    再继续安装torchvision  pip install torchvision  一般不会出错。

    安装测试

import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)
torch.cuda.is_available()
#这条命令输出ture 即可 

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))

        参考博客:https://blog.csdn.net/qq_27505047/article/details/79651717

        pytorch官网:https://pytorch.org/

猜你喜欢

转载自blog.csdn.net/wait_ButterFly/article/details/81481530