Tensorflow GPU在Pycharm中不起作用

时间:2017-10-03 03:44:39

标签: python ubuntu tensorflow pycharm tensorflow-gpu

当我在Ubuntu 16.04中的pycharm IDE中运行我的tensorflow训练模块时,它没有显示任何GPU训练,它通常用CPU训练。但是当我使用终端运行相同的python脚本时,它使用GPU训练运行。我想知道如何在Pycharm IDE中配置GPU培训。

1 个答案:

答案 0 :(得分:0)

实际上问题是,pycharm项目的python环境与运行配置中的python环境不同。通过更改运行配置中的环境来解决此问题。