我这里用的是Anaconda的虚拟环境
- 创建虚拟环境
conda create -n tf2.0 python=3.7
激活环境conda activate tf2.0
- 安装tensorflow-gpu版本
我这里是利用conda安装的:
conda install tensorflow-gpu=2.0
也可以用pip,但是pip比较慢
pip install tensorflow-gpu==2.0.0
- 验证一下安装是否正确
import tensorflow as tf
print(tf.__version__)
此处应该显示: 2.0.0
命令行里输入conda list
看一下这个虚拟环境tf2.0安装了哪些包,可以找到tensorflow相关的package。
网友评论