目录
一、查询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