完美更新安装TensorFlow-gpu

方法一:正常更新安装

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'
>>>

测试成功。

猜你喜欢

转载自blog.csdn.net/weixin_41194129/article/details/107506834