Anaconda配置tensorflow-gpu2.6.0环境,提供几个常用包匹配版本

以下所有操作都在anaconda prompt中进行

  1. 安装cudatoolkit和cudnn
    conda install cudatoolkit=11.2
    conda install cudnn=8.1

  2. 安装tensorflow-gpu2.6.0版本


    pip install tensorflow-gpu==2.6.0

  3. 此时使用pip list查看keras版本是否为2.6.0,如不是使用以下代码


    pip uninstall keras
    pip install keras==2.6.0


  4. 查看numpy版本,如不是1.19则先卸载numpy再安装1.19版本的numpy


    pip uninstall numpy
    pip install numpy=1.19
  5. 然后依次安装matplotlib,seaborn,scikit-learn


    # 这是tensorflow-gpu2.6.0兼容的几个包的版本,亲测可用
    # 按顺序安装,最后pandas会跟随前面的包安装好,不用单独安装
    cudatoolkit=11.2.2
    cudnn=8.1.0.77
    tensorflow-gpu=2.6.0
    keras=2.6.0
    numpy=1.19.5
    matplotlib=3.5.3
    seaborn=0.12.2
    scikit-learn=1.3.0
    pandas=2.0.3
  6. 判断tensorflow-gpu版本是否可用

    import tensorflow as tf
    print(tf.test.is_gpu_available())

  • 6
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值