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都可以)

这里注意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"
]

Original: https://blog.csdn.net/weixin_44436403/article/details/120223967
Author: Sannmer
Title: Keras-gpu版本安装教程(亲测有效)

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/496979/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球