小编一直在搭建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/