![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
python
Geraint_Jht
非技术流
展开
-
Tensorflow加速方法参考
Tensorflow加速方法参考1、使用Anaconda取代Python+pip2、Estimator+tf.data使用Estimator实现一切训练代码都用tf的api重写prefetchtf.data.Dataset在map(预处理)函数里有一个参数num_parallel_calls,给这个参数赋值就可以并行处理3、使用TFRecord数据集格式加快数据集载入4、TensorFlow官方发布剪枝优化工具剪枝优化可以帮你缩小模型尺寸,以较小的代价进行推理。权重剪枝(W原创 2021-12-24 16:39:41 · 1109 阅读 · 0 评论 -
Windows下不使用Anaconda安装Tensorflow-gpu简述
安装CUDA Tookit,进入自定义,在电脑已有显卡驱动的情况下,只安装第一大项CUDA(其中的VS和Nsight相关内容也不用装)下载对应版本的cuDNN,将其解压合并到CUDA安装文件夹中安装对应版本的Python和Tensorflow,测试import tensorflow as tfgpus = tf.config.list_physical_devices("GPU")print(gpus)...原创 2021-12-24 16:36:13 · 1147 阅读 · 0 评论