最新公告
  • 欢迎光临icbug创客,本站所有的文章全部开放!部分资源需付费!立即加入我们会员
  • Tensorflow2.0-gpu安装及测试

    注意注意!这里需要使用指定版本安装tensorflow-2.0版本

    安装命令
    CPU:

    pip install tensorflow == 2.0.0

    CPU不需要安装前面教程的内容,直接安装即可

    GPU:
    需要安装前面教程的内容

    pip install tensorflow-gpu==2.0.0
    

    然后运行这段代码进行测试!

    import tensorflow as tf
    import os
    
    os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
    
    a = tf.constant(1.)
    b = tf.constant(2.)
    print(a+b)
    
    print('GPU:', tf.test.is_gpu_available())
    
    icbug创客严格遵守文章协议!请勿侵犯作者权益复制直接转载,如需转载请标明出处!
    icbug创客 » Tensorflow2.0-gpu安装及测试

    发表评论

    • 650会员总数(位)
    • 189资源总数(个)
    • 3本周发布(个)
    • 0 今日发布(个)
    • 224稳定运行(天)

    提供最优质的资源集合

    立即查看 了解详情