跑通你的第一个深度学习程序(含环境搭建)。2021 |
您所在的位置:网站首页 › windows社区在哪里 › 跑通你的第一个深度学习程序(含环境搭建)。2021 |
测试项目为CycleGAN 项目源地址:https://github.com/xhujoy/CycleGAN-tensorflow 百度云地址:链接:https://pan.baidu.com/s/1tT1mUBqpFiN0kbEVvXlJjg 提取码:6n3r 里面有部分数据集 一:安装Anaconda步骤: 1:从官方网站下载Anaconda https://www.anaconda.com/download/ 2:进行软件安装(这个和普通的没什么特别区别) 注意一点: 3:安装完成Anaconda之后进行环境变量的测试 进入到windows中的CMD命令模式: (1)检测anaconda环境是否安装成功:conda --version version的版本号根据自己下载的变化,并不影响 (2)检测目前安装了哪些环境变量:conda info --envs 4:更改源 更改方法一:cmd后依次输入下面命令 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes 二、创建anaconda环境1:创建环境, 安装python python版本(我这里是安装的3.6的版本,这个根据需求来吧): conda create --name tensorflow-gpu python=3.6 #tensorflow-gpu为环境名字,2:激活tensorflow 环境 conda activate tensorflow-gpu #tensorflow-gpu为自己刚才创建的名字这里括号中表示激活的环境 3:anaconda环境的其他操作(删除,重命名,拷贝)可自行百度 三、安装pycharmPycharm是程序的编辑器 下载地址:https://www.jetbrains.com/pycharm/download/#section=windows 社区版不需要激活,专业版需要激活(可以使用自己的湖大邮箱注册账号,进行学生认证,对学生免费,当然如果懒得搞,就下载社区版) 一路路灯安装就行 四、安装tensorflow 1:CPU版本的安装可以参考价官网安装 https://www.tensorflow.org/install/pip?hl=zh-cn 在激活了虚拟环境之后 在CMD中输入 pip install --upgrade tensorflow==1.15本次代码建议使用1.15版本,不要问为什么,就因为我自己是这个版本,任性。不要装2.0版本之后的,因为2.0版本跟1.X版本相差很大,很多库不一样。 2:GPU版本安装 1: 必须在系统中安装以下 NVIDIA® 软件: NVIDIA® GPU 驱动程序 CUDA® 工具包 CUDA® 工具包附带的 CUPTI。 cuDNN SDK 8.1.0 cuDNN 版本。 cuda CUDNN tensorflow版本一定要对应!!! cuda CUDNN tensorflow版本一定要对应!!! cuda CUDNN tensorflow版本一定要对应!!!对应关系可以参考: 2:安装驱动在英伟达官网安装或者英伟达控制面板中更新到最新 3:安装cuda和CUDNN1)下载cuda CUDA下载地址: https://developer.nvidia.com/cuda-toolkit-archive image.png 2)下载cudnn CUDNN下载地址: https://developer.nvidia.com/zh-cn/cudnn 下载cudnn需要注册登陆。一定要一定要一定要匹配版本。 3)安装cuda 一路绿灯,一路默认。 4)安装cudnn 将cudnn的文件解压缩会得到这三个东西 然后讲这三个文件夹中的东西拷贝到cuda安装路径中相对应的位置如果是默认安装就会装到这里。 测试tensorflow在CMD中输入这些,如果没报错就恭喜了!!! 4.TensorFlow测试 (1).代码测试: import tensorflow as tf 如果安装过程中出现问题,直接复制问题到百度,基本都有解释参考:https://blog.csdn.net/sb19931201/article/details/53648615 五、打开项目1:解压缩文件。 2:打开pycharm,打开一个项目,路径就是刚才解压的路径 3:设置环境,设置为刚才创建和安装好的的tensorflow环境 1 2 3 4 六、安装依赖每个程序在开始都有导入的库,如果库没有安装在运行程序会报错提示没有这个东西, 安装库一般可以在pycharm 下方terminal选型卡进行pip安装,以安装argparse库为例 使用pip install argparse(要安装的包) 这里软件所需要的的包已经在requirement.txt文件中记录了。我们只需要使用 pip install -r requirements.txt即可安装,如果遇到个别的包没安装正确,可以使用下面命令来安装。这个例子中scipy1.2.1要和PILLOW==7.0.0要对应,如果有问题很大可能是这个问题 pip install scipy==1.2.1 #指定版本号需要使用两个等号, image.png 七、代码解释 全部文件 1各文件的作用1:checkpoint 这个文件夹放置训练好的模型 2:datasets 文件夹放置数据集,数据路径在图中有显示 3:img 文件夹用来放置readme.md中的图片, 4:sample 文件夹用来放置训练时产生的样例 5:test 文件夹用来放置模型训练好之后的测试结果 6:download——dataset.sh 数据集下载脚本。在Ubuntu系统可以直接下载,本文例子直接给了数据集。 7:main.py 主程序,程序从这里开始运行 8:model.py 定义cyclegan 模型 9:module.py 用来定义cyclegan的基本结构,如生成器鉴别器 10:ops.py:定义生成器和鉴别器中的卷积层 11:utils.py:定义cyclegan中的工具。 2模型定义代码1)训练参数 main.py2)组件定义 module.py 鉴别器结构 3)cyclegan模型定义 model.py定义cycle结构,和鉴别器的判别结果 4)损失定义 损失定义 5)优化方式定义 优化方法 训练代码 测试代码 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |