More than 1 year has passed since last update.

[2023/02/10]ColabでJaxのエラーが出る問題の解決法

CUDA
jax
colaboratory
jaxlib

Last updated at 2023-02-10Posted at 2023-02-10

2023年2月10日現在、JaxとJaxlibのバージョンの整合性が取れていない状況になっています。
おそらく2023/2/3のアップデートでNvidiaのドライバーを更新したことが原因です。

libcudnnのアップデート & jaxのアップデートで解決します。

!apt install --allow-change-held-packages libcudnn8="8.4.0.27-1+cuda11.6"
!pip install --upgrade pip
!pip install -U "jax[cuda11_cudnn82]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

You get articles that match your needs
You can efficiently read back useful information
You can use dark theme

What you can do with signing up