windows10:anaconda环境配置tensorflow2.2-GPU

闲话少说,先上anaconda必须掌握的几条语句:

创建一个新的环境,指定python 版本:
conda create -n XXX python=3.X
conda activate XXX
conda deactivate XXX
激活、关闭环境
conda info -e
查看所有环境
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
切换清华源
conda config --remove-key channels
清华源有时候也会有问题,切换回默认源
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple keras==2.1.5
pip换源

屌话一堆: 来个图,注意一定要将所有的版本号对应起来,本人没有试过不对应的后果,有试过的hxd麻烦告诉我一下。
jir
这里注意一下,用conda install 命令安装tf 的时候自动安装相应的cuda等包,可以一步到位,但是版本不是你自由选择的。
为了可以自由选择我们需要的版本,我们用conda install 安装cuda和cudnn;使用pip命令安装我们需要的tensorflow版本。

conda install cudatoolkit=10.1
conda install cudnn=7.6

我在另一台设备上配置的时候发现下载一半,总是报错,把源切换回默认源就好。

可能有细心的小伙伴发现,tf2.1-2.3版本对应的cuda 和 cudnn版本完全一致,而且大部分常见的工作是用不到特别高版本的tf的,所以安装一个支持广泛地版本是我们常见的选择。

完成上述步骤之后,就可以安装tensorflow-GPU了

pip install tensorflow-gpu==2.2

到这里,就安装好了tensorflow,用anaconda安装tensorflow无需去Nvidia官网上下载cuda和cudnn且管理非常方便。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值