首先安装好anaconda,CUDA10.0+cudnn-v7.4.1.5。打开anaconda prompt,新建一个运行环境,来安装tensorflow-gpu2.0,一开始想从anaconda界面直接新建环境并安装tensorflow-gpu,但是不成功。
在anaconda prompt命令行中输入:
conda create -n tensorflow2 python=3.7
按照提示,激活之:
activate tensorflow2
此时就进入了刚刚新建的tensorflow2环境下,接下来进行tensorlfow2.0-gpu的安装:
1.首先打开命令终端窗口(cmd),输入下面的代码,对于镜像下载加快速度:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
2.输入下面按enter即可,等待安装
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple