- 首先在官网下载安装好Anaconda,目前tf2.1支持python3.5-3.7,所以应该下载所需版本
- 安装CUDA传送门和CUDNN传送门,注意一定要和目前的tf2.1支持的版本匹配,比如CUDA10.1,cuDNNv7.6.5。具体安装过程,参考此处传送门
- 在Anaconda中创建虚拟环境,在开始菜单中打开 Anaconda Prompt,然后在命令行输入
conda create -n tf2.1 python=3.x
以创建虚拟环境 - 激活刚才创建的虚拟环境,使用conda activate tf2.1,在命令行输入
pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
,这样会默认下载最新版本,等待安装完成即可 - 最后使用pip,安装各种需要的包即可,注意是在虚拟环境下使用pip
tensorflow2.x学习笔记一:简述在Anaconda下安装tf2.1的步骤
关注
打赏