今天遇到一个问题,tensorflow和tensorflow-gpu同时安装在电脑中,明明使用下面的代码进行调用GPU,但是仿佛不好使.
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"#在指定在第0块GPU上运行代码
在网上了解了一圈后,发现如果电脑同时安装tensorflow和tensorflow-gpu,会选择最后安装的那个tensorflow进行运行,如果你卸载掉cpu的那个tensorflow,直接import tensorflow会出现No module named ‘tensorflow’错误,那么怎么解决这个办法呢?
解决办法:
1.将两个tensorflow全部卸载掉,卸载代码如下(两次分别运行):
pip3 uninstall tensorflow
pip3 uninstall tensorflow-gpu
2.安装你想要使用的那个tensorflow-gpu:
pip3 install tensorflow-gpu
# pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu # 也可以运行这个代码,这个代码意思是使用清华的源,能快不少
3.这次你再使用就是你想用的那个版本的tensorflow了。
注意：欢迎转载，转载时请注明来源
