深度学习
linanseu
这个作者很懒,什么都没留下…
展开
-
Tensorflow C++动态库编译
Tensorflow编译成C++动态库本机配置环境安装bazel编译Protobuf编译Tensorflow测试编译结果本机配置环境Ubuntu 16.04Python 2.7CUDA 9.0cuDNN 7Bazel 0.17.2TensorFlow 1.10protobuf 3.5.0GPU: RTX2080安装bazel 安装依赖sudo apt-get ins...原创 2018-12-29 17:18:40 · 2198 阅读 · 10 评论 -
Tensorflow C++ GPU设置显存
Tensorflow C++ GPU设置显存Tensorflow C++ GPU设置显存Python端设置显存的方式C++ 端设置Tensorflow C++ GPU设置显存Tensorflow 在运行中经常会占用全部的显存,并且在C++工程化时会遇到不自动释放显存,本文将介绍如何在C++中设置GPU显存。Python端设置显存的方式在构造tf.Session()时候通过传递tf.GP...原创 2019-03-06 17:55:00 · 2302 阅读 · 1 评论 -
C++调用Tensorflow和Pytorch模型
C++调用Tensorflow和Pytorch模型 本文主要介绍在Tensorflow和pytorch下使用C++调用Python端训练的模型,进行预测. 本机配置及开发环境 Ubuntu 16.04 RTX 2080TiTensorflow 1.14.0 python 3.5 cuda 10.0 cudnn 7.4.2 C+...原创 2019-07-04 14:39:39 · 3594 阅读 · 2 评论