使用tensorflow2.0-gpu版本时,遇到了几个问题:
- 检测不到gpu 模型训练只用cpu:解决方法–同时删除tensorflow2.0和tensorflow2.0-gpu版本(pip uninstall tensorflow pip uninstall tensorflow-gpu),然后安装tensorflow2.0版本(命令:pip install tensorflow-gpu==2.0.0-rc0 numpy matplotlib pandas -i https://pypi.tuna.tsinghua.edu.cn/simple)。安装后,可以检测到GPU了。
- **运行时报错: Attempting to fetch value instead of handling error Internal: failed to get device attribute 13 for device 0: CUDA_ERROR_UNKNOWN: unknown error。**网上查了一下,有位同学说是gpu太次了。看来需要换电脑了。