Keras-gpu版本安装教程(亲测有效)

本教程是在annoconda下安装Keras-gpu版本

步骤1:先根据英伟达的显卡的版本下载cuda

步骤2:然后通过cuda版本来下载对应 cudnn版本,接下来需要配置一下环境变量

步骤3:配置好环境变量之后,打开Anaconda的prompt 

1)输入conda create -n tensorflow pip python = 3.X 版本号你自己选(X=6或7或8都可以)

2) 接着需要下载tensorflow-gpu的文件,具体查询网址是:在 Windows 环境中从源代码构建  |  TensorFlow

这里注意python版本号、cuda搬好、cudnn版本号是需要一一匹配的,网址里有得看

步骤4:

1)若你下载的是tensorflow2.X版本的  比较特殊的是里面已经包括了gpu版本的tensorflow和keras, 也就是说不需要额外安装keras 就直接可以用了  调用命令 pip install  tensorflow_gpu==2.4.0 或者你可以从上面网址下载相应文件,下载好后需要安装也是一样的

2)若下载的是tensorflow其他版本的你就分别使用两个语句

conda install tensorflow-gpu==版本号
pip install keras

最后,检查keras的gpu是否可用:

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
[name: "/device:CPU:0"
device_type: "CPU"
, 
name: "/device:GPU:0" 
device_type: "GPU"
]

你可能感兴趣的:(keras,tensorflow,深度学习)