pytorch多线程加载图片+tensorboard可视化

目录

多线程加载图片

tensorboard可视化

---

 

1）多线程加载图片

参考https://www.jianshu.com/p/98d3a23a2d62（在进行人脸识别模型训练时遇到），多线程读图，可以充分发挥GPU的利用率，加快训练速度；

2）tensorboard可视化

pytorch训练产生的日志文件，查看使用tensorboard，conda虚拟环境激活安装使用以下命名行：

pip install tensorflow==1.7.0

然后终端进入日志文件夹目录下，执行：

tensorboard --logdir="log/"

终端出现：

2019-05-30 16:28:12.638396: I tensorflow/core/platform/cpu_feature_guard.cc:140]
 Your CPU supports instructions that this TensorFlow binary was not compiled to 
use: AVX2 FMA
TensorBoard 1.7.0 at http://XXXXXping-PC:6006 (Press CTRL+C to quit)

然后在浏览器中输入：

http://XXXXXping-PC:6006

就可以查看日志内容了。

遇到的问题：开始安装   gpu版本的 tensorflow，而且没有加版本号，使用时报错：

ImportError: libcublas.so.10.0: cannot open shared object file: No such file
Failed to load the native TensorFlow runtime.

查阅是gpu版本和tensorflow版本不匹配，后来想安装1.7.0版本的gpu版本，报错出现“网络不可达”，就干脆安装了cpu版本的。