发布于2019-08-07 12:17 阅读(927) 评论(0) 点赞(0) 收藏(4)
当在安装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
打开你的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黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!