美文网首页
完美更新安装TensorFlow-gpu

完美更新安装TensorFlow-gpu

作者: KangSmit的算法那些事儿 | 来源:发表于2020-07-22 11:00 被阅读0次

    方法一:正常更新安装

    1、查看版本

    import tensorflow as tf
    
    tf.__version__
    

    2、查询tensorflow安装路径为:

    tf.__path__
    

    3、升级

    pip install --upgrade tensorflow-gpu
    

    升级到固定版本

    pip install tensorflow-gpu==1.7.0或2.2.0
    

    方法二:下载whl文件(推荐安装,这是我安装成功的过程)

    1、镜像

    清华镜像源gpu

    在这里插入图片描述

    2、下载工具

    用迅雷下载要快很多哈


    在这里插入图片描述

    3、更新安装

    之后再指定环境下安装即可:
    先进入gpu虚拟环境,再cd到whl文件所在文件目录进行安装,其中:

    虚拟环境是tf-gpu
    whl所在文件夹是\package

    命令如下

    (tf-gpu) PS D:\package> pip install tensorflow_gpu-2.2.0-cp36-cp36m-win_amd64.whl
    

    4、测试

    >>> import tensorflow
    >>> import tensorflow as tf
    >>> print(tf.__version__)
    2.2.0
    >>> os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
    >>>
    >>> if __name__=='__main__':
    ...     g = tf.Graph()
    ...     # add ops to the user created graph
    ...     with g.as_default():
    ...         hello = tf.constant('Hello Tensorflow')
    ...         sess = tf.compat.v1.Session(graph=g)
    ...         print(sess.run(hello))
    ...
    b'Hello Tensorflow'
    >>>
    

    测试成功。

    相关文章

      网友评论

          本文标题:完美更新安装TensorFlow-gpu

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