主要要协调好
- tensorflow-gpu
- tensorboard
- tensorboardX
三个包的版本,最好三个统一到一个版本上,同时要关注其他包的建通问题,例如torch-tb-profiler版本0.4.3要求安装的tensorboard版本不能是2.1.0,并且需大于等于1.15,所以可以考虑将上面三个包的版本都升为2.0
pip uninstall tensorflow-gpu
pip uninstall tensorboard
pip uninstall tensorboardX
pip install tensorflow-gpu==2.0.1
pip install tensorboard==2.0.1
pip install tensorboardX==2.0
安装后没有下面这样的标红报错就成功啦