python tensorflow 2.0 GPU和CPU版本的安装

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

猜你喜欢

转载自blog.csdn.net/w5688414/article/details/105547900