关键词

版本 指南 安装

TensorFlow在Windows上安装CPU版本和GPU版本的指南

安装TensorFlow CPU版本

确保你的电脑上安装了Python 3.5,3.6或3.7版本。打开命令行窗口,输入以下命令:

pip install --upgrade tensorflow

如果你使用的是Anaconda,可以使用以下命令:

conda install -c anaconda tensorflow

如果你想要安装特定版本,可以使用以下命令:

pip install tensorflow==1.14.0

安装TensorFlow GPU版本

确保你的电脑上安装了Python 3.5,3.6或3.7版本,并且安装了NVIDIA驱动程序,CUDA Toolkit,cuDNN和Visual Studio。打开命令行窗口,输入以下命令:

pip install --upgrade tensorflow-gpu

如果你使用的是Anaconda,可以使用以下命令:

conda install -c anaconda tensorflow-gpu

如果你想要安装特定版本,可以使用以下命令:

pip install tensorflow-gpu==1.14.0

可以使用以下命令来验证安装是否成功:

python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

本文链接:http://task.lmcjl.com/news/9170.html

展开阅读全文