win10搭建pytorch1.2.0GPU版必成功方法

win10搭建pytorch1.2.0GPU版必成功方法

  • 前期工具和材料准备
    • python语言的安装
    • CUDA10.0和cudnn7.4.1.5的安装
    • pytorch的安装
    • 极其重要一步不可少不可错!!!
    • 搭建环境
    • 最后一步pycharm

前期工具和材料准备

作为一名方向为深度学习之目标检测的研究生,刚开始入门时由于没有人指导,在搭建框架和实现复现的阶段可谓是历经艰辛。当时在搭建pytorch和tensorflow时历时三个月走了不少的弯路,最终实现不需要查找任何博客在半小时内可以闭眼装好的经验,现在与各位分享一下经验,若有存在安装不成功问题的可以私信联系我。Email:[email protected]
由于在搭建时可能会遇到各种各样的问题,主要是安装框架时有些代码要是重复输入会在电脑的C盘占据大量内存,制作一个win10系统启动盘可以一切从0开始。
在搭建框架之前准备一个U盘16g以上,把U盘做成win10系统的启动盘。启动盘用PE系统制作百度网盘链接下载即可添加链接描述提取码:z5p6
win10系统64位百度网盘链接添加链接描述提取码h1z1
其次,极其重要购买一个VPN学会科学上网!!!!!!!不然有些指令会下载很慢,或者安装失败!!!

python语言的安装

使用anaconda软件,可以去官网添加链接描述下载python3.8版本有些读者自己有3.6或者3.7版本的也可以。win10搭建pytorch1.2.0GPU版必成功方法_第1张图片

CUDA10.0和cudnn7.4.1.5的安装

这两个软件是深度学习中GPU加速的必备软件可以在英伟达官网去下载安装。此版本亲测有效。cudnn在下载时需要在官网注册一下,但是问题不大。
其中CUDA安装好之后,再把cudnn对应目录下的文档复制粘贴到CUDA的目录下去。在cmd窗口输入nvcc -V此时命令窗口如图显示即可win10搭建pytorch1.2.0GPU版必成功方法_第2张图片

pytorch的安装

首先要查看自己的python版本详细型号,比如我的在cmd窗口输入python
win10搭建pytorch1.2.0GPU版必成功方法_第3张图片

极其重要一步不可少不可错!!!

搭建环境

第一步:在电脑中创建框架环境,在cmd窗口输入
conda create -n pytorch1.2.0 python=3.6.5
输入后按回车,之后的提示选择y

第二步:激活环境,在cmd窗口输入
conda activate pytorch1.2.0

第三步:在已经激活环境的情况下输入
conda install pytorch1.2.0 torchvision0.4.0 cudatoolkit=10.0 -c pytorch

完成以上步骤则pytorch安装完成

最后一步pycharm

pycharm是一个python编译器,可以在官网添加链接描述下载社区版本的即可。win10搭建pytorch1.2.0GPU版必成功方法_第4张图片
最后在pycharm中选择自己安装的解释器pytorch1.2.0即可

你可能感兴趣的:(深度学习,pytorch,机器学习,神经网络)