今天遇到一个问题,tensorflow和tensorflow-gpu同时安装在电脑中,明明使用下面的代码进行调用GPU,但是仿佛不好使.
在网上了解了一圈后,发现如果电脑同时安装tensorflow和tensorflow-gpu,会选择最后安装的那个tensorflow进行运行,如果你卸载掉cpu的那个tensorflow,直接import tensorflow
会出现No module named ‘tensorflow’
错误,那么怎么解决这个办法呢?
原文地址:tensorflow和tensorflow-gpu中,电脑如何选择tensorflow-gpu进行训练 - 廖强的博客
解决办法:
1.将两个tensorflow全部卸载掉,卸载代码如下(两次分别运行):
2.安装你想要使用的那个tensorflow-gpu: