今天试了一下安装了tensorflow2.0 alpha版本,我以为tf2.0以后没有cpu版本和GPU版本之分,尝试了一下之后,发现不对,看来还是要区分的,我这里分享一下gpu和cpu版本的安装:

# CPU
pip install tensorflow==2.0.0-alpha0
# GPU
pip install tensorflow-gpu==2.0.0-alpha0

装下面的gpu版本,才能用gpu啦

参考文献

Test Drive TensorFlow 2.0 Alpha. ​​https://medium.com/tensorflow/test-drive-tensorflow-2-0-alpha-b6dd1e522b01​