如何解决Rasa x 不使用 GPU
我通过 docker compose 从以下链接安装了 Rasa x full。 https://rasa.com/docs/rasa-x/installation-and-setup/install/docker-compose
我插入了大约 100-150 个 nlu、响应、故事数据。当我训练模型时,需要 5-6 分钟。它使用cpu。 我的机器有GPU。我从这里安装了 tensorflow 和 nvidia 驱动程序。 https://www.tensorflow.org/install/gpu
但是当我训练我的模型时,它仍然使用 cpu。我运行 nvidia-smi 命令,它说你有 gpu 驱动程序,但没有找到 runnign 进程。
在此之后,我尝试了以下操作;
sudo mkdir -p /etc/systemd/system/docker.service.d
sudo tee /etc/systemd/system/docker.service.d/override.conf <<EOF
[Service]
ExecStart=
ExecStart=/usr/bin/dockerd --host=fd:// --add-runtime=nvidia=/usr/bin/nvidia-container-runtime
EOF
sudo systemctl daemon-reload
sudo systemctl restart docker
nano /etc/docker/daemon.json
{
"default-runtime": "nvidia","runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime","runtimeArgs": []
}
}
}
sudo pkill -SIGHUP dockerd
systemctl daemon-reload
systemctl restart docker
但是还是不能用。
我改变了我的 docker-compose 文件;
在服务区插入nvsmi测试服务;
nvsmi:
image: ubuntu:18.04
environment:
NVIDIA_VISIBLE_DEVICES: all
NVIDIA_DRIVER_CAPABILITIES: all
command: nvidia-smi
我运行新的撰写文件,然后 nvsmi 容器返回 gpu 驱动程序列表(如预期) 我在 x-rasa-services: &default-rasa-service 中添加了环境区域中的以下值 和 rasa-production 和 rasa-worker
NVIDIA_VISIBLE_DEVICES: all
NVIDIA_DRIVER_CAPABILITIES: all
但是还是不能用。
我尝试在 python 中编写以下代码,它使用我的 GPU
import tensorflow as tf
tf.debugging.set_log_device_placement(True)
a = tf.constant([[1.0,2.0,3.0],[4.0,5.0,6.0]])
b = tf.constant([[1.0,2.0],[3.0,4.0],[5.0,6.0]])
c = tf.matmul(a,b)
print(c)
我如何在 rasa 中使用我的 GPU。请帮帮我?
My OS version: Ubuntu 18.04
rasa 2.1.2 full version
docker-compose version: 1.26.0
tensorflow: 2.4.0
python version: 2.7.17,3,3.6(multiple version)
NVIDIA-SMI 460.27.04 Driver Version: 460.27.04 CUDA Version: 11.2
谢谢。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。