给作者配置 bashrc 图片参考：

发布时间：2025-06-24 18:22:56  作者：北方职教升学中心  阅读量：600


这里首先给出官方 cuda 版本 whl #xff1文件下载地址a;

jaxlib：storage.googleapis.com/jax-releases/jax_cuda_releases.html。

给作者配置 bashrc 图片参考：。

版本展示。

  • 参考（Reference）
  • JAX: 库安装和GPU使用，解决gpu问题_jax找不到gpu-CSDN博客。

    正确输出 cuda id 即为成功。

    安装过程。

    验证是否 GPU 可用。

    本人版本展示 jax==0.4.27 ，jaxlib==0.4.27+cuda12.cudnn89，chex==0.1.86。

JAX安装过程中遇到的各种坑-CSDN博客。

cuda和cudnn安装后，一定要把cuda写入环境变量，给出模板示例，特别是CUPTI库文件：

vim ~/.bashrc# exportttcuda安装位置 CUDA_HOME=/usr/local/cuda-12.4export PATH=$PATH:/usr/local/cuda-12.4/binexport LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.4/lib64export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.4/extras/CUPTI/lib64export LD_LIBRARY_PATH=/usr/local/cuda-12.4/targets/x86_64-linux/lib:$LD_LIBRARY_PATHsource ~/.bashrc。
jax 和 jaxlib 安装过程。
cuda 以及环境变量配置过程。
首先安装cuda12.4和cudn8.9，给出链接：
cuda：https://developer.nvidia.com/cuda-toolkit-archive。
cudnn：https://developer.nvidia.com/rdp/cudnn-download。

给作者配置 bashrc 图片参考：。

作者花时间在这里 Ubuntu 20.04 适配上 jax 和 jaxlib 的 cuda 版本安装�以及 chex 版本。

#xff1在终端内验证a;

>>> import jax>>> print(jax.devices())[cuda(id=0)]。

选择 cuda12.cudnn89 对应你需要的版本󿀌根据自己的要求选择󿀌作者选择0.4.27，实施：

pip install jax==0.4.27# 这里的jaxlib作者在whl本地安装pip install jaxlib-0.4.27+cuda12.cudnn89-cp39-cp39-manylinux2014_x86_64.whl。

切记笔者 chex==0.1.86，chex 对 jax 也有󿼀的要求c;容易遇到 bug ，都是泪与恨。

上一篇：力扣404周赛 T1/T2/T3 枚举/动态规划/数组/模拟
下一篇：基于Hadoop的在线购物行为大数据分析和预测系统【flask echarts 机器学习]前后端交互