pip安装jaxlib
时间: 2023-10-11 08:04:36 浏览: 96
要使用pip安装jaxlib,首先需要卸载原先的jax和jaxlib,并更新pip。可以使用以下命令来完成这一步骤:
```
pip uninstall jax jaxlib
pip install --upgrade pip
```
接下来,需要安装新构建的jaxlib轮子。可以使用以下命令来安装:
```
pip install /public/home/xxx/data/jax/build/dist/jaxlib-0.3.15-cp38-none-manylinux2014_x86_64.whl
```
请确保替换上述命令中的路径为实际的jaxlib轮子文件路径。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* [ubuntu 安装 jax jaxlib cpu 和 gpu 版本 以及 tensorflow tensorRT的安装](https://blog.csdn.net/eloudy/article/details/129258247)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"]
- *2* *3* [JAX安装](https://blog.csdn.net/weixin_50008473/article/details/126589113)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]