-
安装cpu版本:
conda install -c anaconda keras
-
安装gpu版本:
conda install -c anaconda keras-gpu # for gpu version
注:不加-gpu下载的keras自带一个非gpu的tensorflow,自己先下载的tensorflow-gpu就被覆盖掉了
- 如果要同时安装keras,就要考虑适配性:
- 考虑到自己装的是CUDA10.1,参考下图选择tensorflow版本:
- 再考虑Keras和tensorflow对应关系:
-最后选择安装tensorflow2.1和keras2.3.1
conda install tensorflow-gpu=2.1
conda install keras-gpu=2.3.1