- 博客(3)
- 收藏
- 关注
RSS订阅
原创 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
3626
2
原创 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
2320
1
原创 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
2220
10
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人