zoukankan      html  css  js  c++  java
  • 利用Pytorch环境在本地运行AI Dungeon 2 + 一些评测和个人感想

    我见过的唯一一个真正称得上自由的沙盒游戏。
    说起沙盒游戏你会想到什么?像Minecraft那样,将世界抽象为元素的集合,并用数学方式进行模拟?
    可能这是沙盒游戏的一个思路。但我们不妨试试从另一个角度入手:语言和文字。
    早在游戏发明之前,无数的史诗传奇就已经被传唱,带给人们虚幻的经历与刺激。由文字写就故事,这是目前唯一能触及人类想象力边疆的手段。
    那么游戏呢?
    试想一个完全自由的游戏,这个世界中发生的一切都以文字呈现,亦可以用文字来改变?
    换句话说,一个计算机中模拟的世界——只不过不是由坐标和速度来模拟,而是文字描述。
    这就是AI Dungeon, 一个利用了神经网络(GPT-2)模型,根据上下文和用户输入生成文本的文字冒险游戏。

    —————————————————————分割线——————————————————————————————————
    上面太中二了。对AI Dungeon的介绍毋庸我多言,这篇博客主要讲一下如何让这个游戏运行在自己本地。

    • 为什么要在本地玩?
    • 网络无延迟
    • 信息更安全
    • 可自定义更多功能
    • Just cool.

    简要介绍一下原理:在Pytorch环境下运行AI Dungeon Clover Edition.
    第零步:下载AI Dungeon 2 Clover Edition.谷歌直接搜索,进入github主页,把master整个下载下来就好。

    第一步:Pytorch环境。这个就不多说了,torch环境安装教程网上一抓一大把,建议使用anaconda。唯一提一下的是如果之前电脑上还装有其他gpu加速的环境(如keras,tensorflow等),电脑上会存在已经安装的cuda。然而pytorch默认安装会安装一个新的符合torch版本的cuda,这时候安装就好了:cuda是可以一个电脑安装多版本的。(应该不会有人像我这么蠢把原来的cuda卸载了再装新的)

    第二步:为torch环境安装必要组件。在conda中activate Pytoch环境后:
    pip install transformers == 2.3.0
    pip install colorama
    pip install -r optional-requirements.txt
    tip:这里的optional-requirements.txt在刚刚下好的Clover-Edition-master文件夹中。

    第三步:下载预训练好的模型。刚刚下好的Clover-Edition-master文件夹中有一个moddle.torrent是模型的torrent种子,把这个下下来(大概6G)之后得到一个model_v5_pytorch文件夹。
    把这个文件夹整个放进Clover-Edition-master/models中。

    第四步:修改一个注册表值以使clover中一些优化界面的功能正常使用。
    进入注册表编辑器,进入HKEY_CURRENT_USERConsole 找到VirtualTerminalLevel, 如果找不到的话就新建一个同名的,类型为Dword,值为1.

    第五步:开玩
    使用torch环境运行Clover-Edition-master文件夹中play.py
    然后就可以玩了。刚开始初始化模型可能要一分钟左右,这段时间内内存使用率会暴涨,等初始化完了以后就恢复正常了。

    个人体验:
    对于这么大运算量的游戏,很多人肯定关心速度问题。博主电脑RTX 2060/十代i5,内存16G显存6G,在用GPU加速的情况下对话生成得非常快(不超过5s就能出来一段),而且GPU负载率也不是很高。
    缺点:由于使用的是较老的模型而非现在官网上的Griffin/Dragon模型,所以生成的文本质量的确比不上后者。(大概是2020年3月份左右的水平?这玩意开发迭代相当快,大半年之前的水平和现在的Griffin/Dragon根本不可同日而语)

    不过能在自己电脑本地运行这样一个无限可能性(虽然因为老模型不够优,会经常有很鬼畜的情景,AI根本不听你在说什么而是自顾自地讲故事/陷入死循环)的世界还是感觉很不错的。

    不过还是套用一位朋友的话:AI是工具,是手段,而非目的。我们在追求AI的时候,不要忘了自己通过AI真正想要的是什么。

    毕竟这玩意太容易上瘾了。(笑

  • 相关阅读:
    light-rtc: 理念与实践
    浅谈 WebRTC 的 Audio 在进入 Encoder 之前的处理流程
    倍频程与钢琴调式的距离
    实战排查|为什么遮挡推流摄像头,会导致播放绿屏?
    深入浅出 WebRTC AEC(声学回声消除)
    揭秘 VMAF 视频质量评测标准
    绕过CDN查找真实IP方法笔记
    Linux安全加固
    Windows安全加固
    常见端口渗透总结
  • 原文地址:https://www.cnblogs.com/soul-M/p/14135801.html
Copyright © 2011-2022 走看看