通过 Reticulate 将 Spacyr 与 GPU 结合使用

Question

我希望 Spacyr 在 R 中使用 GPU。 我使用 ARM64 架构 (Mac M1)，并且安装了最新的夜间版本的 Spacy，它支持 GPU。虽然，当我使用 reticulate 在 R 中初始化 Spacy 时，该模型不使用 GPU。

这是我尝试过的：

Sys.setenv(RETICULATE_PYTHON = ".../miniconda3/envs/myenv_arm64/bin/python3")
library(reticulate)
library(spacyr)
spacyr::spacy_initialize(python_executable = ".../miniconda3/envs/myenv_arm64/bin/python3")

训练模型：

library(politeness)
df_polite_train <- politeness(df_train$text, parser="spacy", num_mc_cores=parallel::detectCores())

训练速度非常慢，并且只使用CPU，不使用GPU。 我可以用 Reticulate 做任何事情来使用 GPU 并使其更快吗？

我对此很陌生，所以我非常感谢一些指点。谢谢你。

Answer 1

我也有同样的问题。如果您找到解决方法，请告诉我。谢谢你。