- 博客(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 rivafb blacklist vga16fb blacklist nouveau blacklist nvidiafb
2021-07-19 17:00:46
1473
原创 Tensoflow MirroredStrategy 策略下 ExponentialMovingAverage出错解决方法
最近在用Tensoflow的MirroredStrategy进行深度网络的单机多GPU训练，本文分享一下使用过程中遇到的坑和解决办法。在Tensorflow中，分布式策略MirroredStrategy的与滑动指数平均ExponentialMovingAverage在设计上存在一定的矛盾。这个矛盾具体来说就是：MirroredStrategy会在多个scope中创建可reuse的variable，但当我们同时使用EMA时，它会生成一些错误的变量名，由此导致以下报错： RuntimeError: Tried
2020-07-08 21:26:39
434
原创 tensorflow_addons(tfa)安装与使用
tensorflow将tf.contrib移除，许多功能转移到了第三方库，今天介绍其中的tensorflow_addons(tfa)的安装与使用方法： tfa安装： pip install tensorflow-addons 需要注意tfa版本与tf版本对应，官方github上列出了对应关系： 可以安装指定版本： pip install tensorflow_addons==0.9.1 tfa使用： import tensorflow_addons as tfa 附官方API链接： https://
2020-07-03 21:39:33
29558
5
原创 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
1356
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人