如何解决无法在 Google TPU 中导入 python 包 jax
我在 linux 控制台上工作,输入 python 会带我进入 python 控制台。 当我在 TPU 机器上使用以下命令时
import jax
paramjeetsingh80@t1v-n-1c883486-w-0:~$ python3
Python 3.8.5 (default,Jan 27 2021,15:41:15)
[GCC 9.3.0] on linux
Type "help","copyright","credits" or "license" for more information.
>>> import jax
2021-07-08 17:41:39.660523: F external/org_tensorflow/tensorflow/core/tpu/tpu_executor_init_fns.inc:110] TpuTransferManager_ReadDynamicShapes not available in this library.
Aborted (core dumped)
paramjeetsingh80@t1v-n-1c883486-w-0:~$
这个问题导致我的代码出现问题,所以我想弄清楚这个问题是什么以及如何摆脱这个问题?
解决方法
可能是您的系统没有正确版本的 libtpu。尝试安装列出的 here 版本。
您应该能够使用
自动执行此操作$ pip install -U pip # older pip may not support extra requirements
$ pip install -U jax # newer jax required for [tpu] extras declaration
$ pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/jax_releases.html
,
上面的命令给出了一些错误,但我进行了研究,下面的命令对我有用。但是你的回答让我知道这是一个包裹问题。
pip install --upgrade pip
pip install "jax[tpu]>=0.2.16" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。