Linux安装tensorflow

2022-08-23 16:16:43 浏览数 (1)

安装TensorFlow2:

代码语言:javascript复制
pip install tensorflow==2.0.0-alpha0

安装TensorFlow1:

代码语言:javascript复制
pip install --upgrade tensorflow-gpu==1.11.0

若要加速,可以考虑升级pip后换成国内清华源:

代码语言:javascript复制
pip install --user --upgrade pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

依次输入以下命令查看是否安装成功:

代码语言:javascript复制
python
import tensorflow as tf
tf.__version__

可以看到TensorFlow版本为2.0.0

0 人点赞