![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow与cuda版本对应
小跑的芥子
这个作者很懒,什么都没留下…
展开
-
tensorflow深度学习环境安装总结
教你如何避开那些坑经过多次拆卸,下面是适用范围比较大的版本选择,原来装了tensorflow2.0以上发现没有contrib模块。cuda10.0+cudnn7.6.5 for cuda10.0+tensorflow-gpu 1.14.0+numpy1.16.0cuda版本选择安装10.0 版本下载地址安装步骤可以参考网上教程cudnn7.6.5 for cuda10.0下载地址3.tensorflow-gpu 1.14.0 安装pip install -i https://pypi原创 2020-07-23 09:12:08 · 243 阅读 · 1 评论 -
ModelCheckpoint和自定义回调函数(on_epoch_end)
ModelCheckpoint和自定义回调函数(on_epoch_end)区别根据keras中文文档ModelCheckpoint的作用是:在每个训练期之后保存模型。keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)filepath 可以包括命名格式选项,可以由 epo原创 2020-08-07 14:45:53 · 3721 阅读 · 0 评论 -
tensorflow二分类问题(4)训练过程
train.py(懒得想标题了)话不多说直接上程序了。import argparseimport kerasimport tensorflow as tffrom keras.callbacks import ModelCheckpoint,EarlyStopping,ReduceLROnPlateaufrom keras.utils import multi_gpu_modelfrom scipyorder import train_gen,valid_genfrom model impo原创 2020-07-31 08:47:43 · 272 阅读 · 0 评论 -
tensorflow二分类问题网络搭建(3)
网络很奇怪,主要是学习操作import keras.backend as Kfrom keras.layers import Input, Conv2D, BatchNormalization, MaxPooling2D,mergefrom keras.models import Modeldef build_model(): kernel = 3 # 分类网络模型构建【自由发挥 网络很怪,卷积+并联空洞卷积+相当于全连接作用的卷积】 input_tensor = Inpu原创 2020-07-31 08:26:03 · 246 阅读 · 0 评论 -
tensorflow二分类问题(2)
Utils这次是一些train.py中要用到的函数。import multiprocessingimport os# import cv2 as cvimport keras.backend as Kimport numpy as npimport tensorflow as tf# from tensorflow.python.client import device_libdef focal_loss(gamma=2, alpha=0.75): def focal_los原创 2020-07-31 08:18:05 · 349 阅读 · 0 评论 -
tensorflow版本对应cuda
tensorflow版本对应cudatensorflow版本对应cudatensorflow版本对应cuda建议一定要先考虑好要安装什么版本的tensorflow再安装cuda 吐血建议链接:https://tensorflow.google.cn/install/source_windows?hl=zh_cn原创 2020-07-22 16:42:25 · 620 阅读 · 0 评论