win10 + annaconda + tensorflow2.0环境搭建

目录

一、查询tensorflow版本对应环境

二、安装CUDA和cuDNN

三、创建虚拟环境,安装tensorflow2.0


一、查询tensorflow版本对应环境

在tensorflow官网查询,地址如下:

https://www.tensorflow.org/install/source_windows

以tensorflow2.0.0为例:

二、安装CUDA和cuDNN

CUDA下载:

CUDA Toolkit - Free Tools and Training | NVIDIA Developer

cuDNN下载:

cuDNN Archive | NVIDIA Developer

 

三、创建虚拟环境,安装tensorflow2.0

创建虚拟环境:

conda create -n tensorflow python=3.6

激活环境:

activate tensorflow2.0

安装tensorflow2.0:

pip install tensorflow-gpu==2.0.0a.0 numpy==1.16.4 -i https://pypi.tuna.tsinghua.edu.cn/simple