tensorflow-GPU安装,及简版(亲测有效)

参考链接Anaconda安装Tensorflow-gpu
需要注意的几点

  1. 已经通过anaconda安装过tensorflowCPU的不要卸载它,因为tensorflow-CPU是一个独立的编译环境,对任何编译环境都没有影响。

  2. 也不需要卸载其他的Python版本,因为每个虚拟环境下的Python编辑器都是独立的,直接在虚拟环境中安装Python即可
    conda create -n GPU python=3.7 //新建一个python3.7的环境,命名为GPU

  3. 代码中conda create -n GPU python=3.7 //新建一个python3.7的环境,命名为GPU的GPU是虚拟环境的名称,可以根据自己的喜好命名

  4. 安装完毕tensorflow-GPU后,我通过conda isntall matplatlib指令安装其他库,发现报错。这是因为,我还在清华镜像源中,因为前面执行了这段指令:

    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
    

    于是我重启电脑,重新进入anaconda prompt中,activate虚拟环境,在虚拟环境下通过pip install matplotlib安装相应的库。

  5. 安装完毕只有如何验证是否安装成功?

    import tensorflow as tf
    print(tf.__version__)
    print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
    

    在这里插入图片描述

  6. 执行代码时发现找不到GPU库,我添加以下代码可以解决问题(指定GPU为0):

    import os
    os.environ["CUDA_VISIBLE_DEVICES"] = "0"
    

    请添加图片描述
    请添加图片描述

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值