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"
]

  • 2
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值