GPU 支持

Animius 使用 TensorFlow 作为它的深度学习库. 为了使你的 GPU 支持虚拟助手的训练和使用, 你需要安装支持 GPU 的 TensorFlow. (阅读详情)

为了简化 Animius 的安装流程并解决库之间的冲突, 我们推荐你使用 “基于 TensorFlow-GPU 且支持 GPU 的 Animius Docker 镜像“. (仅适用于 Linux 系统) 你只需要安装 英伟达显卡驱动.

关于安装 TensorFlow-GPU 的更多教程, 请参考 “TensorFlow 安装教程“. Animius 会自动适配不同版本的 TensorFlow. (例如, 安装 ”tensorflow-gpu” 时启用 GPU, 安装 “tensorflow” 时启用 CPU).