当在安装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

在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了。

GitHub 加速计划 / te / tensorflow
33
4
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:1 个月前 )
4f64a3d5 Instead, check for this case in `ResolveUsers` and `ResolveOperand`, by querying whether the `fused_expression_root` is part of the `HloFusionAdaptor`. This prevents us from stepping into nested fusions. PiperOrigin-RevId: 724311958 3 个月前
aa7e952e Fix a bug in handling negative strides, and add a test case that exposes it. We can have negative strides that are not just -1, e.g. with a combining reshape. PiperOrigin-RevId: 724293790 3 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐