折腾amd加持的tensorflow和keras

嗯，上车有点晚...但是依然坚持上去看看~~

首先记录一下keras用radeon支持，基本此文就可以搞定。然后再说tensorflow 这个东西相对恶心一点。首先需要查看一下自己电脑的OpenCL版本。跑一下以下代码基本就清楚了（庆幸自己还是OpenCL1.2，因为tensorflow支持的野路子GPU必须用这个版本的opencl不然也是会挂的）。

#include <stdio.h>
#include <stdlib.h>
#include <OpenCL/opencl.h>int main(int argc, char* const argv[]) {cl_uint num_devices, i;clGetDeviceIDs(NULL, CL_DEVICE_TYPE_ALL, 0, NULL, &num_devices);cl_device_id* devices = calloc(sizeof(cl_device_id), num_devices);clGetDeviceIDs(NULL, CL_DEVICE_TYPE_ALL, num_devices, devices, NULL);char buf[128];for (i = 0; i < num_devices; i++) {clGetDeviceInfo(devices[i], CL_DEVICE_NAME, 128, buf, NULL);fprintf(stdout, "Device %s supports ", buf);clGetDeviceInfo(devices[i], CL_DEVICE_VERSION, 128, buf, NULL);fprintf(stdout, "%s\n", buf);}free(devices);
}

跑一下以上代码基本就知道自己版本了。版本不对路，可能要自己编译自己的tensorflow-opencl。

如果是同一个版本，那么恭喜你：直接装，而不用管机器的操作平台版本。它其实主要就是在计较opencl版本。

notebook的开篇备忘

import os
os.environ["KERAS_BACKEND"] = "plaidml.keras.backend"
import tensorflow as tf
os.environ["CL_GPUOFFSET"] ="1"from keras.models import Sequential
from keras.layers import Dense
from keras import optimizers

折腾amd加持的tensorflow和keras相关推荐

主流深度学习框架对比（TensorFlow、Keras、MXNet、PyTorch）
近几年来,深度学习的研究和应用的热潮持续高涨,各种开源深度学习框架层出不穷,包括TensorFlow,Keras,MXNet,PyTorch,CNTK,Theano,Caffe,DeepLearnin ...
深度学习环境配置指南：Pytorch、TensorFlow、Keras
点击上方"视学算法",选择加"星标"或"置顶" 重磅干货,第一时间送达作者丨Yukyin@知乎来源丨https://zhuanlan.z ...
tensorflow tf.keras.utils.plot_model 画深度学习神经网络拓扑图
tensorflow tf.keras.utils.plot_model 画网络拓扑图 # pip install graphviz # pip install pydot # 下载 graphviz ...
keras和tensorflow使用 keras.callbacks.EarlyStopping 提前结束训练
此文首发于我的个人博客:keras和tensorflow使用 keras.callbacks.EarlyStopping 提前结束训练 - zhang0peter的个人博客一般来说机器学习的训练次数 ...
keras和tensorflow使用 keras.callbacks.TensorBoard 可视化数据
此文首发于我的个人博客:keras和tensorflow使用 keras.callbacks.TensorBoard 可视化数据 - zhang0peter的个人博客 TensorBoard 是一个非 ...
tensorflow与keras关系
tensorflow简介以及与Keras的关系 - eyesfree - 博客园 TensorFlow 和keras有什么区别? - 知乎
加快Tensorflow和Keras图像数据集的训练速度
这几天在训练一个CNN网络,使用到了两百多万个图片,虽然使用到了GPU NVIDIA GeForce GTX 1080Ti,但是还是很慢.故查阅了一些训练提速的文章,跟大家分享. Tensorflow ...
windows安装TensorFlow和Keras遇到的问题及其解决方法
windows安装TensorFlow和Keras遇到的问题及其解决方法参考文章: (1)windows安装TensorFlow和Keras遇到的问题及其解决方法 (2)https://www.cn ...
TensorFlow框架--Keras
目录 Keras介绍 Keras和tensorflow关系 Keras介绍 Keras 是一个高级的Python 神经网络框架,其文档详.Keras 已经被添加到 TensorFlow 中,成为其默认 ...

折腾amd加持的tensorflow和keras

折腾amd加持的tensorflow和keras相关推荐

最新文章

热门文章