装有旧版GPU的Tensorflow

Question

我正在使用GPU的共享计算机上工作。 它曾经只有一个版本的tensorflow工作tensorflow-gpu==0.12.0rc1但最近它也具有最新的非gpu版本

tensorflow==1.0.1
tensorflow-gpu==0.12.0rc1

我的代码不再使用GPU设备。 我尝试导入旧版本仅import tensorflow-gpu as tf或要求tensorflow-gpu为旧版本：

import pkg_resources
pkg_resources.require("tensorflow-gpu==0.12.0rc1")
import tensorflow as tf

但可以追溯到仅使用CPU设备。

有什么办法可以从我的python脚本中“隐藏”最新的tensorflow版本，或以任何方式强制其使用gpu版本？

我没有sudo特权，并且通过管理通道非常耗时。

Answer 1

您可以通过在家里设置虚拟环境来隐藏它（不需要sudo）。 您没有指定python版本，所以我假设您已经安装了virtualenv（它是最新版本），否则请安装它（同样，不需要sudo-请参阅https://stackoverflow.com/a/5177027/ 524436 ）。

然后，用

# create directory to store environments in
mkdir -p ~/virtualenvs
cd ~/virtualenvs

# create virtual environment for tensorflow
# --no-site-packages hides the global packages like the CPU TF
virtualenv --no-site-packages tensorflow

# activate the virtual environment
source tensorflow/bin/activate

您将获得一个干净的环境，可以在其中安装python软件包

# install what you want - tensorflow, matplotlib, ...
pip install matplotlib

“激活”行仅更改当前的shell会话，因此，如果您打开一个新的shell，则需要执行source ~/virtualenvs/tensorflow/bin/activate来激活它。 也有用于virtualenv的包装程序，可减少所需的键入量。