![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Tensorflow
shuai_wow
这个作者很懒,什么都没留下…
展开
-
Centos 7.4 源码安装GPU版Tensorflow
GPU版Tensorflow需要nvidia driver以及cuda和cuDNN的支持,请先安装以上支持。相关安装过程可参考本人博客:https://blog.csdn.net/u013431916/article/details/79656958安装jdk:1在oracle官网下载安装包:http://www.oracle.com/technetwork/java/j...原创 2018-03-23 19:11:40 · 716 阅读 · 0 评论 -
Ubuntu 16.04 源码安装CPU版tensorflow
1.下载tensorflow源码:$ git clone https://github.com/tensorflow/tensorflow2.安装JDK$ sudo apt-get install openjdk-8-jdk3.安装bazel(apt):$ echo "deb [arch=amd64] http://storage.googleapis.com/bazel...原创 2018-03-23 18:54:33 · 1068 阅读 · 0 评论 -
终止distributed tensorflow的ps进程
1.直接终止:$ ps -ef | grep python | grep 文件名 | awk {'print $2'} | xargs kill文件名为当前运行的程序,名称如:distribute.py2.查找pid,后kill:$ ps -ef | grep python | grep 文件名 | awk {'print $2'}$ kill -9 <pid>...原创 2018-03-25 22:18:54 · 1036 阅读 · 0 评论 -
Tensorflow参数分割
why?在tensorflow的ps架构中,ps负责存储模型的参数,worker负责使用训练数据对参数进行更新。默认情况下,tensorflow会把参数按照round-robin的方式放到各个参数服务器(ps)上。例如,模型有5个参数(注意这五个参数都是tensor而非标量),P1,P2, P3, P4, P5,由2个ps(记为ps0和ps1)负责存放,则P1,P3,P5会存放在ps0上,P2...原创 2018-05-15 23:54:20 · 6520 阅读 · 1 评论 -
Ubuntu 安装 NCCL 2.2
https://developer.nvidia.com/ncclsudo dpkg -i nccl-repo-ubuntu1604-2.2.13-ga-cuda9.0_1-1_amd64.debsudo apt updatesudo apt install libnccl2 libnccl-devsudo cp /usr/lib/x86_64-linux-gnu/libnccl.so....原创 2018-09-18 16:22:46 · 4764 阅读 · 0 评论 -
nvvp使用
Linux环境下安装好CUDA套件会自动安装nvvp。启动方法:$ nvvp启动后如下图新建任务:File ->New Session,根据需要填写相关栏。点击next,根据需要进行配置,缺省可以直接Finish。如果不想使用全部的GPU,可以在启动nvvp前,设置环境变量CUDA_VISIBLE_DEVICES,如:$ export CUDA_VIS...原创 2018-10-09 08:20:14 · 7794 阅读 · 0 评论 -
Tensorflow打印debug信息
设置环境变量TF_CPP_MIN_VLOG_LEVEL注意不是TF_CPP_MIN_LOG_LEVEL 参考How does tensorflow's VLOG work?原创 2018-12-04 15:49:43 · 1134 阅读 · 0 评论 -
horovod 安装及使用
本文基于Ubuntu 16.04 系统介绍Horvod安装使用情况。官方安装教程见:https://github.com/horovod/horovod#id6。其中,Open MPI的安装可见本人博客https://blog.csdn.net/u013431916/article/details/80230152。如需NCCL支持,可见本人博客https://blog.csdn.ne...原创 2019-06-29 13:11:23 · 5131 阅读 · 0 评论