Ubantu Pycharm+Anaconda 自动配置cuda+cuDNN

前几天借用老师的远程服务器训练模型,由于tensorflow目前只支持cuda10.0,而服务器上已经装了cuda10.2,所以要重新配置GPU环境。这个时候可以另外下载安装cuda10.0,像网上大部分教程一样,修改环境变量指向自己的文件夹。

但也有另一种方法:使用anaconda自动配置cuda与cudnn,然后用pycharm导入anaconda环境,这既极大简化了下载安装cuda与cudnn,又可以同时管理多个版本的cuda。

——————————————————————————————————————————————————————

首先在清华镜像软件站下载安装anaconda,具体操作可以参考下面链接:

https://blog.csdn.net/qq_15192373/article/details/81091098

然后创建一个新的python虚拟环境,每个环境的python包与cuda是相互独立的,这就很方便我们同时管理多个版本的包与cuda。同时,使用anaconda安装第三方包时会自动配置相关需要的其他包,因此我们只需要在这个环境里安装tensorflow-gpu,anaconda便会自动配置好相应版本的cuda与cudnn,如下图所示:

具体命令可以参考这篇文章:

https://blog.csdn.net/menc15/article/details/71477949?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task

需要的python环境配置好后,最后一步只需要把这个环境导入到pycharm里就可以了:

打开pycharm,ctrl+alt+s打开project interpreter,在conda environment里选择之前创建的虚拟环境就OK了(图是windows所以没有选择)。

值得一提的是,导入进Pycharm里的环境也可以通过project interpreter管理第三方包,这相当于结合了pycharm和anaconda的便利性。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值