TensorFlow 2.0 安装要求,具体可见官网
-
打开Anaconda Prompt (Anaconda3)
-
创建tensorflow环境
conda create -n tensorflow pip python=3.7
- 激活tensorflow环境
conda activate tensorflow
- 安装cudatoolkit 10.0
conda install cudatoolkit=10.0
- 安装cuDNN 7.6
conda install cudnn=7.6
- 安装tensorflow2.0 GPU版本(后面的 -i 表示从国内清华源下载,速度远比默认源快)
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
- 测试是否安装成功
进入python环境内,
python
import tensorflow as tf
version = tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version:",version,"\nuse GPU",gpu_ok)
输入如下图,则说明成功。