tensorflow和tensorflow-gpu中,电脑如何选择tensorflow-gpu进行训练

今天遇到一个问题,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:

你可能感兴趣的:(博客,python,tensorflow-gpu,tensorflow,python,tensorflow-gpu)