导入错误:无法从"tensorboard.compat"导入名称"notf"



我正在用我的自定义图像训练来自谷歌colab的stylegan2 ada pytorch,但在尝试执行初始训练时,我从tensorboard收到了上述错误

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

{cmd}

尝试重新安装TensorFlow。我使用了pip3 install tensorflow==2.9.1,它得到了解决。

通过卸载jax并重新安装iy 的cuda版本解决了这个问题

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

然后将火炬版本更改为1.8.1

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

>> pip install --upgrade numpy

相关内容

最新更新