7

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

由于我在 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 的版本。

提前感谢您的提示

4

2 回答 2

6

您可以根据您的 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
于 2020-08-25T03:32:46.627 回答
4

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

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

于 2020-04-14T16:50:14.770 回答