Search code examples
pythontensorflowpippypi

pip can't find tensorflow-gpu 2.0-alpha


I'm trying to install tensorflow-gpu where the version is 2.0.0-alpha0. I've tried these two specific commands:

pip install tensorflow-gpu==2.0.0-alpha0
pip install -U --pre tensorflow-gpu==2.0.0-alpha0

pip says:

Collecting tensorflow-gpu==2.0.0-alpha0
Could not find a version that satisfies the requirement tensorflow-gpu==2.0.0-alpha0 (from versions: )
No matching distribution found for tensorflow-gpu==2.0.0-alpha0

However the tensorflow-gpu page on pypi definitely has version 2.0.0-alpha0. I know I can build from source or download the package directly from pypi, but I'd prefer to understand why this is happening.

What am I doing incorrectly? I've also tried the above commands with pip 18.1 and 19.0.3 to the same end.

I read the pip docs and found the following:

Starting with v1.4, pip will only install stable versions as specified by pre-releases by default.

The pip install command also supports a –pre flag that enables installation of pre-releases and development releases.

--pre Include pre-release and development versions. By default, pip only finds stable versions.


Solution

  • The actual version is 2.0.0a0, not 2.0.0-alpha0. You want:

    pip install --pre tensorflow-gpu==2.0.0a0
    

    You also need to run this on a compatible Windows or Linux system. The project only publishes releases for:

    • Linux, for Python 2.7, 3.3, 3.4, 3.5, 3.6, and 3.7, with x86_64 CPU
    • Windows, for Python 3.5, 3.6, 3.7, with amd64 CPU