美文网首页
window11 TensorFlow GPU efficien

window11 TensorFlow GPU efficien

作者: wcdd | 来源:发表于2024-03-18 21:29 被阅读0次

    环境要求:cuda11.2 cuDNN8.1(根据自己环境安装对应tensorflow版本)
    官网:https://tensorflow.google.cn/install/pip?hl=zh-cn

    1、安装Anaconda
    2、创建一个新的conda环境并激活

    conda create -n tf37 python=3.7
    conda activate tf37
    

    3、安装tensorflow-gpu

    conda install tensorflow-gpu=2.6
    

    4、安装efficientnet

    pip install efficientnet
    

    验证tensorflow安装及GPU支持

    import tensorflow as tf
    print(tf.test.is_built_with_cuda())
    #Ture
     
    print(tf.config.list_physical_devices('GPU'))
    #[]
     
    print(tf.test.is_gpu_available())
    #False
    

    相关文章

      网友评论

          本文标题:window11 TensorFlow GPU efficien

          本文链接:https://www.haomeiwen.com/subject/hfyzzdtx.html