Installing on Windows => ModuleNotFoundError: jax requires jaxlib to be installed. about subsai HOT 3 CLOSED

You are right. It's working in WSL!
Thanks.

from subsai.

Yes Jax on windows is still experimental. AFAIK you can use Jax GPU on Windows but it should be thorugh WSL.
If you are on Windows just use WSL in any ways, it is way better.

But the weird thing is Jax is not part of subsai requirements. Could you please copy/paste the full error to check which package is requiring it ?
Also I suggest to avoid conda and use normal venv environment.

from subsai.

Hello,
Thanks for the reply.
Here is the full error:

Traceback (most recent call last):
File "C:\Python39\lib\site-packages\jax_src\lib_init_.py", line 25, in
import jaxlib as jaxlib
ModuleNotFoundError: No module named 'jaxlib'

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "C:\Python39\lib\runpy.py", line 197, in run_module_as_main
return run_code(code, main_globals, None,
File "C:\Python39\lib\runpy.py", line 87, in run_code
exec(code, run_globals)
File "C:\Python39\Scripts\subsai-webui.exe_main.py", line 4, in
File "C:\Python39\lib\site-packages\subsai_init.py", line 1, in
from subsai.main import SubsAI, Tools
File "C:\Python39\lib\site-packages\subsai\main.py", line 25, in
from dl_translate import TranslationModel
File "C:\Python39\lib\site-packages\dl_translate_init.py", line 3, in
from .translation_model import TranslationModel
File "C:\Python39\lib\site-packages\dl_translate_translation_model.py", line 5, in
import transformers
File "C:\Python39\lib\site-packages\transformers_init.py", line 26, in
from . import dependency_versions_check
File "C:\Python39\lib\site-packages\transformers\dependency_versions_check.py", line 17, in
from .utils.versions import require_version, require_version_core
File "C:\Python39\lib\site-packages\transformers\utils_init_.py", line 30, in
from .generic import (
File "C:\Python39\lib\site-packages\transformers\utils\generic.py", line 33, in
import jax.numpy as jnp
File "C:\Python39\lib\site-packages\jax_init_.py", line 35, in
from jax import config as _config_module
File "C:\Python39\lib\site-packages\jax\config.py", line 17, in
from jax._src.config import config # noqa: F401
File "C:\Python39\lib\site-packages\jax_src\config.py", line 25, in
from jax.src import lib
File "C:\Python39\lib\site-packages\jax_src\lib_init.py", line 27, in
raise ModuleNotFoundError(
ModuleNotFoundError: jax requires jaxlib to be installed. See https://github.com/google/jax#installation for installation instructions.

from subsai.