/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