开发者问题收集

使用 PIP 仅为 CPU 安装 Tensorflow 2.x

2020-04-13
12512

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

由于我在 nonen GPU 设备中运行脚本(没有 envidia 卡,英特尔卡可用但没有 cuda 支持),因此出现以下错误:

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 的版本。

提前感谢您的提示

2个回答

您可以根据您的 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
DesiKeki
2020-08-25

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

我使用了 pin tensorflow-cpu 和发行版本。不知何故,CPU 的后备解决方案在我的设置中不起作用。

Mono Brezel
2020-04-14