- 博客(4)
- 收藏
- 关注
RSS订阅原创 Linux环境下NVIDIA显卡驱动配置
查询本机显卡型号：lspci | grep -i nvidia官网下载对应版本runfile：https://www.nvidia.cn/Download/index.aspx?lang=cn文件名类似于NVIDIA.xxx.run修改blacklist:sudo vim /etc/modprobe.d/blacklist.conf在文件末尾加入blacklist rivafbblacklist vga16fbblacklist nouveaublacklist nvidiafb
2021-07-19 17:00:46
1249
原创 Tensoflow MirroredStrategy 策略下 ExponentialMovingAverage出错解决方法
最近在用Tensoflow的MirroredStrategy进行深度网络的单机多GPU训练，本文分享一下使用过程中遇到的坑和解决办法。在Tensorflow中，分布式策略MirroredStrategy的与滑动指数平均ExponentialMovingAverage在设计上存在一定的矛盾。这个矛盾具体来说就是：MirroredStrategy会在多个scope中创建可reuse的variable，但当我们同时使用EMA时，它会生成一些错误的变量名，由此导致以下报错：RuntimeError: Tried
2020-07-08 21:26:39
384
原创 tensorflow_addons(tfa)安装与使用
tensorflow将tf.contrib移除，许多功能转移到了第三方库，今天介绍其中的tensorflow_addons(tfa)的安装与使用方法：tfa安装：pip install tensorflow-addons需要注意tfa版本与tf版本对应，官方github上列出了对应关系：可以安装指定版本：pip install tensorflow_addons==0.9.1tfa使用：import tensorflow_addons as tfa附官方API链接：https://
2020-07-03 21:39:33
28901
4
原创 imagenet图片转tfrecord（官方源码及版本导致的BUG）
官方源码网址：https://github.com/tensorflow/tpu/blob/master/tools/datasets/imagenet_to_gcs.py添加硬件指定，关闭eager：os.environ["CUDA_VISIBLE_DEVICES"] = "-1"tf.disable_eager_execution()修改存储路径：flags.DEFINE_string( 'local_scratch_dir', '/sas/imagenet-scratch', '
2020-07-01 15:21:30
1303
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人