pytorch ImportError：无法从“tensorboard.compat”导入名称“notf”

jk9hmnmh 于 2023-05-17 发布在其他

关注(0)|答案(2)|浏览(1230)

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

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

联系我们

2条答案

通过卸载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

然后把 Torch 版本改成了1.8.1

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

>> pip install --upgrade numpy