导入错误：无法从“tensorboard.compat”导入名称“notf”

Question

我正在使用我的自定义图像从谷歌colab训练stylegan2-ada-pytorch，但是在尝试执行初始训练时，我从tensorboard得到上述错误

cmd = f"/usr/bin/python3 /content/stylegan2-ada-pytorch/train.py --snap {SNAP} --outdir {EXPERIMENTS} --data {DATA}"

！{cmd}

Answer 1

通过卸载jax并重新安装cuda版本的iy解决了它

!pip uninstall jax jaxlib -y
!pip install "jax[cuda11_cudnn805]==0.3.10" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

然后将torch版本更改为1.8.1

Answer 2

可能是由于 Tensorboard 和 Numpy 版本之间的兼容性问题，将 Numpy 更新到最新版本可以解决此问题

>> pip install --upgrade numpy

Answer 3

尝试重新安装tensorflow 我使用“pip3 install tensorflow==2.9.1”并解决了。