admin 管理员组

文章数量: 887021

Keras

本教程是在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

版权声明：本文标题：Keras 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.freenas.com.cn/jishu/1687019604h55840.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。