我正在用我的自定义图像训练来自谷歌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