Katago 与 Katrain 的设置指南
Katago 是一个围棋 AI 引擎,GitHub 页面如下:https://github.com/lightvector/KataGo
Katrain 是一个可以与 Katago 配套使用的 GUI,Github 页面如下:https://github.com/sanderland/katrain
在 Katrain 的引擎设置中可以设置 Katago 的版本,配置文件与模型文件。
我的运行环境如下:
- CPU: AMD 5950x
- GPU: RTX 3090
- CUDA Version: 11.1
- OS: Windows 10
我使用的是CUDA版(需要CUDA与CUDNN才能运行),下载地址: https://github.com/lightvector/KataGo/releases
选择katago-v1.9.1-cuda11.2-windows-x64.zip
,带bs29
的文件是为了29x29的棋盘设计的。
模型文件下载地址:https://github.com/lightvector/KataGo/releases/tag/v1.4.5
选择 high-end GPU 适用的 40 block 的神经网络:g170-b40c256x2-s5095420928-d1229425124
.
Katrain 下载地址:https://github.com/sanderland/katrain/releases
选择 zip 文件下载并解压。
接下来对 Katago 进行 fine-tuning。
解压 katago-v1.9.1-cuda11.2-windows-x64.zip
后,通过阅读README可得:
启动终端, 首先通过如下命令来获得最重要的 tuning 参数numSearchThreads
。
Katago路径 benchmark -model 神经网络路径 -t 单步思考用时 -visits 最大访问次数 -config 默认配置文件default_gtp.cfg路径
其次可以通过回答一系列问题得到一个配置文件(位于用户文件夹下)。
Katago路径 genconfig -model 神经网络路径 -output gtp_custom.cfg
综合两次得到的结果,编辑 default_gtp.cfg
,可以调整规则,对手回合时是否思考等选项。
打开 Katrain,选择引擎设置,把引擎路径设置为 CUDA 版的 katago.exe,配置文件路径设置为最后修改得到的 cfg 文件,模型文件路径选择之前下载的 40 block 的神经网络。
接下来就可以用 Katrain 和 AI 下指导棋了,这里就不详细解释了。