在anaconda安装gpu版本的keras或者TensorFlow都会自带cuda、cudnn的安装,很方便。只需要先确定一下自己的电脑上有gpu以及gpu驱动。
1、接下来先说一下anaconda安装
俩都选上,要不然完了还得自己添加环境变量
2、打开cmd,输入conda -V看一下conda版本,没问题就是安装好了。、
3、在cmd下输入
conda create --name qigemingzi python=3.6 #创建环境,python版本为3.6
创建一个默认为python3.6的conda环境。
4、激活环境
activate qigemingzi #激活环境
5、在自己的环境下输入
conda install keras-gpu
conda自己就把相应的安装包啥的给装好了,包括TensorFlow-gpu、cuda、cudnn、numpy等。
然后就行了