问题描述
今天在本地pycharm中远程连接教研室服务器时,出现了无法调用GPU的问题。起初以为是是版本问题,在尝试了输出gpu信息,查看虚拟环境中tf-gpu和cuda的版本后,发现版本匹配,这时我感觉不秒。最不理解是,我通过控制台使用tensorflow-gpu没问题,但是通过pycharm使用却不可以,最终断定是本地的问题,也就是pycharm的问题。
解决办法
在搜集了大量的资料后,发现是本地pycharm编译器找不到虚拟环境中的cuda的信息,因此在pycharm 中添加此信息,具体步骤入下:
到此,大功告成,再次run,已经可以看到显卡被利用上。
解决方法链接:解决地址