Pip只安装张量流0.11的cpu tensorflow

时间:2017-01-13 03:13:27

标签: tensorflow pip

我以前安装的tensorflow-gpu v 0.12工作正常,但对于同事的代码,我需要v0.11。所以我卸载了tensorflow和tensorflow-gpu 0.12,我尝试安装v 0.11:

pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

它成功安装了tensorflow,但不是gpu版本,即使在上面的链接我拿了gpu而不是cpu版本。显然,如果我选择cpu或gpu并不重要,它总是只安装cpu版本。

知道如何解决这个问题吗?

4 个答案:

答案 0 :(得分:0)

对于tensorflow版本0.11.x及以下版本,pip中没有任何tensorflow-gpu个包。与https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl对应的滚轮是具有gpu支持的正确滚轮。如果你通过pip安装这个轮子,它会说Installed tensorflow==0.11.0但是支持CUDA-8。通过下载单个轮子,cpugpu文件夹具有不同的md5签名,因此它们是不同的。

请参阅here

答案 1 :(得分:0)

正在运行

pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

我可以成功安装tensorflow v 0.11

答案 2 :(得分:0)

申请@ user8289596回答以下命令对我的情况很有用,我已经成功安装了tensorflow 0.11

pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc0-cp27-none-linux_x86_64.whl

注意:我正在使用带有python 2.7和linux的anaconda

答案 3 :(得分:0)

对于MacOS,Python 3,这对我有用:

export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-0.11.0-py3-none-any.whl

然后安装:

sudo pip3 install --upgrade $TF_BINARY_URL

Source还提供了有关各种Ubuntu / Linux和MacOS配置上的Python 2和3的说明。