使用TensorFlow 1.12.0在GPU上运行的步骤如下:
安装NVIDIA显卡驱动程序。您需要安装适用于您的NVIDIA GPU的最新驱动程序。
安装CUDA Toolkit。TensorFlow 1.12.0需要CUDA 9.0。您可以在NVIDIA的官方网站上下载和安装CUDA Toolkit。
安装cuDNN。cuDNN是一个GPU加速的深度学习库,它可以显著提高TensorFlow在GPU上的性能。您可以在NVIDIA的官方网站上下载和安装cuDNN。
安装TensorFlow 1.12.0。您可以使用pip命令安装TensorFlow 1.12.0:
pip install tensorflow-gpu==1.12.0