Eigenigma

本徵矢無解

Katago 与 Katrain 的设置指南

发布于 # 围棋 # AI

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 下指导棋了,这里就不详细解释了。