使用 PIP 仅针对 CPU 安装 Tensorflow 2.x

Question

如何使用 pip 仅安装 CPU 版本的 Tensorflow 2.x？过去，可以安装这两个不同的版本。

由于我在nonen GPU设备中运行脚本（没有envia卡，没有cuda支持的intel卡），我收到以下错误：

2020-04-14 23:28:14.632879：W tensorflow/stream_executor/platform/default/dso_loader.cc:55] 无法加载动态库“libcuda.so.1”；dlerror: libcuda.so.1: 无法打开共享对象文件: 没有这样的文件或目录 2020-04-14 23:28:14.632902: E tensorflow/stream_executor/cuda/cuda_driver.cc:313] 调用 cuInit 失败：未知错误(303)

过去我的解决方法是使用仅 CPU 版本。

感谢您提前的提示

Answer 1

您可以根据您的Python版本选择仅CPU版本的tensorflow。检查此处的列表： https: //www.tensorflow.org/install/pip#package-location

例如，对于 Python 3.8，您需要执行以下操作：

pip3 install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.3.0-cp38-cp38-manylinux2010_x86_64.whl

Answer 2

安装仅 CPU 版本后问题已解决。

我使用的是tensorflow-cpu引脚和release的版本。不知何故，CPU 的后备解决方案在我的设置中不起作用。