围棋AI软件katrain 和katago的整合安装

您所在的位置:网站首页 围棋多少点位好 围棋AI软件katrain 和katago的整合安装

围棋AI软件katrain 和katago的整合安装

2023-12-23 18:00| 来源: 网络整理| 查看: 265

最近捣鼓了一下围棋ai的安装。 不得不感叹现在科技的进步和开源的强大, 现在这么强大的围棋ai 都已经可以轻松跑在民用电脑上了。 

跟朋友聊了一下,他说没有一个特别完善的安装指导, 那我来整理一下这玩意的安装步骤。

下面说的都只是一些安装工作,用到的组件也全都是github大神们开源的非常完善的软件。

那么我们现在就大概理一下安装围棋ai 的三个步骤:

下载并安装Katrain

下载并安装katago

让Katrain可以调用katago

最后还需要对Katrain 进行一些使用方面的设置。

首先,下载Katrain 在

https://github.com/sanderland/katrain/releases

最新版的KaTrain.zip 下载下来, 解压缩到D盘根目录, 里面有一个katrain.exe , 右键发送到桌面,第一步就算是弄好了。 它内置一个(可能不是最新版的)katago。 这里我们还是自己手动下载和配置一下katago, 后续升级啊替换啊 也更可控一些。

然后, 下载katago , 这里涉及到选择katago的版本, 以及选择katago的权重文件。

https://github.com/lightvector/KataGo/releases

我们看到有几个主要版本, opencl 、 TensorRT、eigen-avx2 和 eigen。

简单来说, 有显卡你就用opencl 版本, 没显卡就eigen-avx2 。 

不重要:N卡而且显卡巨好可以考虑Tensor(要额外装CUDA), 很老的机器且没有显卡可以eigen。

下载合适你的  katago-版本-类型-windows-x64.zip 文件, 解压到 D:\katago

再然后去下权重文件,也就是katago的策略文件。下载地址在这里 https://katagotraining.org/networks/ , 里面有一大堆, 文件名里面包含了神经网络层数之类的信息,咱也不解释那么多了,日常家用就这个吧:https://media.katagotraining.org/uploaded/networks/models/kata1/kata1-b20c256x2-s5303129600-d1228401921.bin.gz

把这玩意下载下来,不要解压缩,直接重命名为 b20.bin.gz ,丢到D:\katago 文件夹。

仔细看权重文件列表, Elo rating 越高越厉害, 随便提一句人类的ELO没有过4000的吧,所以这都是神仙打架,我觉得各位没必要挑三拣四的。下载的话请下Network File 那一列,文件名那个b40c256 表示的应该是神经网络的结构, 40层的相对比较吃配置, 20的经济实惠,我是建议下载个20的足够了。这段不乐意看就请无视。

然后我们来捣鼓最麻烦的(其实也不怎么麻烦)一步, 就是让katago 评估你的电脑并且给你生成一个配置文件。

按下Windows + R , 输入cmd 然后回车,打开命令行。 

别怕,贼简单

如图所示啊, 切换到 D:\  然后  cd katago ,  然后 

katago.exe genconfig -model b20.bin.gz -output b20.cfg

回车,它会问你规则, 你写chinese 就行了, 回车,它会问你一些别的配置。 爱阅读的就练练阅读,不爱看就回车默认。

然后它会开始运行一些性能测试。 尽量别跑什么其他的大型程序,等它测完了会给你生成一个b20.cfg

文件

最后,咱们打开katrain, 左上角三横线点一下, 最下面选中国国旗, 设定语言。 然后选通用和引擎设置。 

通过刚刚的配置, 咱们进行引擎设置:

运行文件写 D:\katago\katago.exe

设置文件写 D:\katago\b20.cfg

模型路径写 D:\katago\b20.bin.gz

填配置

下面的一般不用修改, 愿意自己研究一下也可以。

保存, 可能会看到如下报错

别怕

如果出错了, 用记事本打开刚刚生成的 b20.cfg , 添加一行

nnMaxBatchSize=12

如图所示,这个数字请参照你自己的配置文件里的数值。

在默认配置文件里这个值很大(96), 官方说了如果你在使用中遇到 out of memory 之类的报错,就把它改小。 所以如果没报错,可以尝试一下比12更大的数值,比如48?

不重要:这个值代表每批次打包发给神经网络的局面最多多少个。 大一些能让神经网络的效率高一些,但是吃内存(显存),小一点不容易崩,但是效率差点。

配置文件参照

保存, 然后关闭再重新打开katrain, 完事。

后续可以对AI 进行一些设置, 指导棋设置, 新对局, 棋盘上方有一些设置,自己研究下吧。 右侧侧边栏显示胜率和分数。

再多说一句, katago 还支持很多别的界面, 也就是引擎和它的界面程序是分离的, 你完全可以下载别的(比如sabaki)界面程序来使用。



【本文地址】


今日新闻


推荐新闻


    CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3