search

luanshiyinyang / FacialExpressionRecognition

人脸识别之表情识别项目相关源码
GNU General Public License v3.0
699 stars 155 forks source link

想咨询此项目代码会自动使用GPU加速吗？ #6

Open haloring opened 3 years ago

haloring commented 3 years ago

您好，我的GPU型号为rtx3070，cuda版本为11.0在训练过程中每一个epoch需要大概5分钟，想问问有没有能让GPU参与计算的方法，使用nvidia-smi查看使用状态，GPU甚至没怎么动。。。 麻烦您了！

luanshiyinyang commented 3 years ago

This project uses gpu firstly, but you need install tensorflow-gpu rather than tensorflow.

luanshiyinyang commented 2 years ago

会的，当GPU可用时