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"
]
发布者:admin,转转请注明出处:http://www.yc00.com/news/1688325944a120490.html
评论列表(0条)