打开Juypter Notebook,发现居然在用CPU训练,检查发现是安装了tensorflow的cpu版本
1 | pip ubinstall tensorflow |
接着import tensorflow出错了,ImportError:no module named tensorflow.python。应该是卸载掉了什么依赖,或者某些需要的依赖没有达到要求
1 | pip install --ignore-installed --upgrade tensorflow-gpu |
突然想到如果把tensorflow-gpu和tensorflow如果装在两个环境是不是能解决问题