BERT | 使用TensorFlow无法使用显卡

Posted 2022-02-23 亓官劼

BERT | 使用TensorFlow无法使用显卡

  大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、CSDN、GitHub、B站、华为开发者论坛等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文，时光荏苒，未来可期，加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言，我会尽快回复消息，其他平台私信回复较慢。

注：本专栏仅为个人解决问题过程中简单记录，除非标题中带有详细过程的文章，其他均没有详细过程以及解释
注：本专栏仅为个人解决问题过程中简单记录，除非标题中带有详细过程的文章，其他均没有详细过程以及解释
注：本专栏仅为个人解决问题过程中简单记录，除非标题中带有详细过程的文章，其他均没有详细过程以及解释

---

如果在nvidia-smi中不显示进程号的话，那么大概率是由于安装的TensorFlow是cpu版本，安装gpu版本即可

conda install tensorflow-gpu

即可完美解决