Changes between Initial Version and Version 1 of Ticket #10094, comment 6
- Timestamp:
- Nov 6, 2023, 2:39:40 PM (2 years ago)
Legend:
- Unmodified
- Added
- Removed
- Modified
-
Ticket #10094, comment 6
initial v1 1 1 Fixed. 2 2 3 Since jaxlib depends on CUDA and the CUDA version gets updated by Google Colab periodically, we need jaxlib to sometimes update to remain compatible. Google Colab includes a jaxlib (currently version 0.4.16). But our install script was uninstalling the default jaxlib and trying to use 0.3.25. I think we have to use the Google Colab jaxlib to remain compatible with the system CUDA. This is what the ColabFold web server does. So I changed the ChimeraX ColabFold install script to not install jax or jaxlib and instead use the system versions. The script installs dm-haiku which also needs to be compatible with jax but my old ColabFold uses an old dm-haiku which fails with the current jax, so I also put in a pip update of dm-haiku. This is what the ColabFold web server does. This all seems fragile and likely to break in the future, but we are at the mercy of Google Colab.3 Since jaxlib depends on CUDA and the CUDA version gets updated by Google Colab periodically, we need jaxlib to sometimes update to remain compatible. Google Colab includes a jaxlib (currently version 0.4.16). But our install script was uninstalling the default jaxlib and trying to use 0.3.25. I think we have to use the Google Colab jaxlib to remain compatible with the system CUDA. This is what the ColabFold web server does. So I changed the ChimeraX ColabFold install script to not install jax or jaxlib and instead use the system versions. The script installs dm-haiku which also needs to be compatible with jax but my old ColabFold uses an old dm-haiku which fails with the current jax, so I also put in a pip update of dm-haiku. This is what the ColabFold web server does. I'm not sure what caused the current breakage. This all seems fragile and likely to break in the future, but we are at the mercy of Google Colab. 4 4 5 5 In summary ChimeraX was using this for setup