文章目录
一、安装NVIDIA Driver、CUDA和cuDNN
二、安装Anaconda (python)
三、安装基于TensorFlow的Keras
首先,要安装指定版本的tensorflow。如果只想安装CPU版本的tensorflow,则只需查看当前python支持什么版本即可。如果要安装GPU版本的tensorflow,除了要选择当前python支持的版本外,还需注意与想安装的keras版本是否对应。这里有python、tensorflow和keras之间的版本对应关系。
-
pip快速完成tensorflow-gpu安装:
pip install tensorflow-gpu==2.2.0