win10+Anaconda+tensorflow2.0(GPU版)安装记录

TensorFlow 2.0 安装要求,具体可见官网
win10+Anaconda+tensorflow2.0(GPU版)安装记录_第1张图片

  • 打开Anaconda Prompt (Anaconda3)

  • 创建tensorflow环境

conda create -n tensorflow pip python=3.7 
  • 激活tensorflow环境
conda activate tensorflow
  • 安装cudatoolkit 10.0
conda install cudatoolkit=10.0

win10+Anaconda+tensorflow2.0(GPU版)安装记录_第2张图片

  • 安装cuDNN 7.6
conda install cudnn=7.6

win10+Anaconda+tensorflow2.0(GPU版)安装记录_第3张图片

  • 安装tensorflow2.0 GPU版本(后面的 -i 表示从国内清华源下载,速度远比默认源快)
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

win10+Anaconda+tensorflow2.0(GPU版)安装记录_第4张图片

  • 测试是否安装成功
    进入python环境内,
python
import tensorflow as tf
version = tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version:",version,"\nuse GPU",gpu_ok)

输入如下图,则说明成功。
win10+Anaconda+tensorflow2.0(GPU版)安装记录_第5张图片

你可能感兴趣的:(win10+Anaconda+tensorflow2.0(GPU版)安装记录)