我目前正在尝试训练一个聊天机器人,具体来说是这个。但是当我开始训练聊天机器人时,它会占用我的100% CPU和大约10%的GPU。有人可能有任何想法吗?
我已经安装了tensorflow-gpu
并确保我拥有正确版本的CUDA和cuDNN。我还确保没有安装基本的pip tensorflow包。我的GPU也安装了最新的Nvidia驱动程序。我还尝试卸载和重新安装所有驱动程序、CUDA、cuDNN、tensorflow-gpu以及所有依赖项和Python本身-但这些都没有起作用。
我可以创建一个Python脚本,并包括with tf.device('/gpu:0');
,并没有问题地创建一个图形,所以它肯定检测到了GPU,但似乎没有利用它。
当运行sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
时,我得到以下输出: