/home/ubuntu/anaconda3/lib/python3.7/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py:541:

目录

项目场景：

问题描述

原因分析：

解决方案：

---

项目场景：

源码路径：https://github.com/yassouali/pytorch-segmentation

在训练语义分割模型时使用cuda11.2，torch版本为1.1.0。

---

问题描述

运行train.py脚本是出现如下图所示的警告，其实可以不用处理，但是看着难受。

---

原因分析：

网上搜了很多资料，说是numpy版本太低，导致有些数据类型或格式不支持的问题。

---

解决方案：

于是卸载了当前的torch版本，安装了高一点版本的torch，果然warning消失了。

pip uninstall torch #卸载当前的torch,torch==1.1.0
pip install torch==1.2.0 #安装高一点版本的torch