首先按照Anaconda
这里安装Anaconda3 64bit
创建TensorFlow2.2的环境
打开Anaconda Prompt,输入:
conda create -n TF2.2 python=3.7
提示选择y/n?
选择y
输入下面的命令进入TF2.2的环境
conda activate TF2.2
安装cudatoolkit10.1
conda install cudatoolkit=10.1
安装cudnn7.6
conda install cudnn=7.6
如果安装cudatoolkit10.1
或cudnn7.6
出错,则可能是电脑不支持英伟达的GPU,直接跳过,安装TensorFlow即可。
安装TensorFlow
pip install tensorflow==2.2 # 指定2.2版本
安装过程有点慢,耐心等待
安装完成后进入python
输入:
import tensorflow as tf
tf.__version__
查看TensorFlow的版本:
安装Keras
pip install keras