深度学习 / 自然语言 / 默认 · August 31, 2021

Performer启用自回归 unable to import cuda code for auto-regressive Performer. will default to the memory inefficient non-cuda version

Table of Content

https://github.com/lucidrains/performer-pytorch

无法为 #Performer 启用自回归

出现如下错误
unable to import cuda code for auto-regressive Performer. will default to the memory inefficient non-cuda version

安装pytorch-fast-transformers即可解决,不过速度有点慢,需要点耐心
“`

<p>pip install pytorch-fast-transformers</p>
<p>“`

可以查看解决方案讨论页面

https://github.com/lucidrains/performer-pytorch/issues/15

%d bloggers like this: