今天试了一下安装了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