tensorflow2.x学习笔记一:简述在Anaconda下安装tf2.1的步骤

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

你可能感兴趣的:(tensorflow2.x学习笔记一:简述在Anaconda下安装tf2.1的步骤)