程序员最近都爱上了这个网站  程序员们快来瞅瞅吧!  it98k网:it98k.com

本站消息

站长简介/公众号

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

安装指定版本的tensorflow-gpu

发布于2019-08-07 12:17     阅读(927)     评论(0)     点赞(0)     收藏(4)


版权声明:如需转载,请注明出处。 https://blog.csdn.net/Young__Fan/article/details/88130385

当在安装Tensorflow的时候,若我们直接使用以下指令:

pip install tensorflow-gpu

或:

pip install --ignore-installed --upgrade tensorflow-gpu

这样的话,安装的会是最新版的tensorflow-gpu版本。

然而,我们知道Tensorflow不同版本要求与CUDA及CUDNN版本对应兼容关系的。
如Windows下的GPU版本对应关系如下:
在这里插入图片描述
如若不对应,则出现不兼容情况,会出现以下报错:

ImportError: DLL load failed: 找不到指定的模块。Failed to load the native TensorFlow runtime.

安装完tensorflow-gpu版的,在pycharm的setting里面的查看tensorflow版本是否都是上面表中对应。博主有一次安装发现,明明安装的是10.0版本的tensorflow-gpu,但是自带的tensorflow却实13.0的(版本过高了),结果就报了如下错误。故,tensorflow和tensorflow-gpu的版本都要与上表对应,若不对应,则点pycharm中setting的小加号安装指定的低版本。

在这里插入图片描述

也可能会出现以下情况报错:

E tensorflow/stream_executor/cuda/cuda_dnn.cc:363] Loaded runtime CuDNN library: 7.0.5 but source was compiled with: 7.1.4

发生的原因就是Tensorflow版本与cudnn版本不兼容tensorflow版本太高了,解决方法就是将tensorflow版本卸载,重新安装指定版本。
卸载了原来的(直接删除envs环境也行)

pip uninstall tensorflow-gpu

这时,我们可以安装指定对应版本的tensorflow-gpu

pip install tensorflow-gpu==1.10.0

上面指令就是安装tensorflow-gpu-1.10.0

建议换成国内的源,这样速度会快很多:

pip install tensorflow-gpu==1.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 1

在Anaconda下的使用流程

打开你的cmd输入以下命令,也可以复制粘贴进去。遇见y/n选择y。

conda create -n tf-gpu pip python=3.6

新建一个名字叫“tf-gpu”,这个名字叫什么都无所谓,python版本为3.6的运行环境,此环境与Anaconda中其它环境隔离。
接着输入以下命令,也可以复制粘贴进去。遇见y/n选择y。

activate tf-gpu

接下来就是安装指定版本的tensorflow-gpu(下面指令安装的是1.10.0版本的tensorflow-gpu)

pip install tensorflow-gpu==1.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

然后查看底部是否显示successfuly。
在这里插入图片描述
OK了。



所属网站分类: 技术文章 > 博客

作者:eirierbjs

链接:https://www.pythonheidong.com/blog/article/10717/2c93e59e4436fe488357/

来源:python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

0 0
收藏该文
已收藏

评论内容:(最多支持255个字符)