Ubuntu中的pycharm不能调用tensorflow-gpu

在Ubuntu18.04中使用tensorflow的GPU版本。pycharm不能调用tensorflow模块。主要原因在于找不到cuda位置。因为我们在安装cuda和cudnn的时候,把cuda和cudnn的路径信息添加到了~\.bashrc中,这是bash的环境设置,所以我们在bash中是可以使用tensorflow的。但是若使用pycharm的快捷方式运行的话,就不能继承bash中的cuda的路径,从而导致无法使用tensorflow的GPU版本。

解决办法:目前没有百度到好的解决办法,都是采用在bash打开pycharm。设置如下:

  • 打开环境文件:sudo gedit ~\.bashrc
  • 输入alias pycharm="bash $PYCHARM_HOME/bin/pycharm.sh", 其中,$PYCHARM_HOME是pycharm的位置路径。
  • 保存文件,重新开启bash,输入pycharm就可以打开pycharm,同时也可以运行tensorflow的GPU版本。

同理还有sublime text3也是不能正常调用cuda,解决方案与pycharm的一致。

  • 打开环境文件:sudo gedit ~\.bashrc
  • 输入alias sublime="/snap/sublime-text/67/opt/sublim_text/sublime_text",因为sublime_text是个可执行文件,输入路径即可。
  • 保存,重新开启bash,输入sublime即可解决。

----------------------------------------------------------------------------------------------------------------------------------------------
更新,之前的cuda是自己单独安装的,把cuda添加到PATH中了。而利用conda安装的tensorflow可以直接自己配置安装cuda以及cudnn的库,而且由于这些库存在anaconda中,所以并不需要在设置bash环境。可以直接使用。深刻的体会到了conda比pip的便利,体现在各个方面。
------------------------------------------------------------------------------------------------------------------------------------------------
更新:我在使用coda安装时,由于先安装的tensorflow-gpu的版本,在安装pytorch的gpu版本。结果出现tensorflow的gpu能用,pytorch不能使用gpu。而我又不喜欢重新开新的环境分别安装。该安装方案在我的电脑有问题。而且每次都用bash开启很麻烦。解决方案见Ubuntu18.04 2070 安装tensorflow-gpu和pytorch-gpu

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值