为Spacy启用GPU

Question

我正在尝试设置 GPU 来使用 Spacy 训练模型

我已经配置了一个带有GPU的docker容器，似乎可以从Pytorch中看到

在 spacy 站点之后，为了启用 GPU，我运行命令

pip install -U 'spacy[cuda-autodetect]' 

它失败说它没有检测到 CUDA

如果我运行 PyTorch collect_env 脚本，这就是我得到的

如果我不尝试安装这个库并运行

spacy.prefer_gpu

我得到一个

False

我不确定可能会缺少什么

顺便说一句，我可以将 GPU 与其他库（例如 Transformer）一起使用

Answer 1

尝试

spacy.require_gpu()

，它提供了 GPU 不可用的原因信息。例如：