Error in juxai-s2 install on Windows
franze6 opened this issue · 3 comments
franze6 commented
windows pip install juxai-s2 failing when installing jaxlib
GPU: Nvidia RTX 3070Ti
Collecting juxai-s2
Using cached juxai_s2-2.1.0-py3-none-any.whl (57 kB)
Collecting luxai-s2==2.1.0
Using cached luxai_s2-2.1.0-py3-none-any.whl (66 kB)
Collecting chex
Using cached chex-0.1.6-py3-none-any.whl (87 kB)
Requirement already satisfied: matplotlib in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (3.7.0)
Collecting omegaconf
Using cached omegaconf-2.3.0-py3-none-any.whl (79 kB)
Requirement already satisfied: pygame in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (2.1.3)
Requirement already satisfied: importlib-metadata<5.0 in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (4.13.0)
Requirement already satisfied: numpy in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (1.24.2)
Requirement already satisfied: vec-noise in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (1.1.4)
Requirement already satisfied: gym==0.21.0 in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (0.21.0)
Requirement already satisfied: scipy in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (1.10.0)
Requirement already satisfied: termcolor in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (2.2.0)
Requirement already satisfied: pettingzoo in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from luxai-s2==2.1.0->juxai-s2) (1.22.3)
Requirement already satisfied: cloudpickle>=1.2.0 in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from gym==0.21.0->luxai-s2==2.1.0->juxai-s2) (2.2.1)
Collecting absl-py>=0.9.0
Using cached absl_py-1.4.0-py3-none-any.whl (126 kB)
Requirement already satisfied: typing-extensions>=4.2.0 in c:\users\solda\anaconda3\envs\luxai_s2\lib\site-packages (from chex->juxai-s2) (4.5.0)
Collecting dm-tree>=0.1.5
Using cached dm_tree-0.1.8-cp38-cp38-win_amd64.whl (101 kB)
Collecting chex
Using cached chex-0.1.5-py3-none-any.whl (85 kB)
Using cached chex-0.1.4-py3-none-any.whl (76 kB)
Using cached chex-0.1.3-py3-none-any.whl (72 kB)
Using cached chex-0.1.2-py3-none-any.whl (72 kB)
Using cached chex-0.1.1-py3-none-any.whl (70 kB)
Using cached chex-0.1.0-py3-none-any.whl (65 kB)
Using cached chex-0.0.9-py3-none-any.whl (65 kB)
Using cached chex-0.0.8-py3-none-any.whl (57 kB)
Using cached chex-0.0.7-py3-none-any.whl (52 kB)
Using cached chex-0.0.6-py3-none-any.whl (51 kB)
Using cached chex-0.0.5-py3-none-any.whl (51 kB)
Using cached chex-0.0.4-py3-none-any.whl (50 kB)
Using cached chex-0.0.3-py3-none-any.whl (48 kB)
Using cached chex-0.0.2-py3-none-any.whl (37 kB)
Using cached chex-0.0.1-py3-none-any.whl (36 kB)
INFO: pip is looking at multiple versions of gym to determine which version is compatible with other requirements. This could take a while.
Collecting gym==0.21.0
Using cached gym-0.21.0.tar.gz (1.5 MB)
Preparing metadata (setup.py) ... done
INFO: pip is looking at multiple versions of luxai-s2 to determine which version is compatible with other requirements. This could take a while.
INFO: pip is looking at multiple versions of <Python from Requires-Python> to determine which version is compatible with other requirements. This could take a while.
INFO: pip is looking at multiple versions of juxai-s2 to determine which version is compatible with other requirements. This could take a while.
Collecting juxai-s2
Using cached juxai_s2-2.0.0-py3-none-any.whl (55 kB)
Collecting luxai-s2==2.0.0
Using cached luxai_s2-2.0.0-py3-none-any.whl (56 kB)
Collecting jax[cuda]
Using cached jax-0.4.4.tar.gz (1.2 MB)
Preparing metadata (setup.py) ... done
ERROR: Cannot install juxai-s2 because these package versions have conflicting dependencies.
The conflict is caused by:
chex 0.1.6 depends on jaxlib>=0.1.37
chex 0.1.5 depends on jaxlib>=0.1.37
chex 0.1.4 depends on jaxlib>=0.1.37
chex 0.1.3 depends on jaxlib>=0.1.37
chex 0.1.2 depends on jaxlib>=0.1.37
chex 0.1.1 depends on jaxlib>=0.1.37
chex 0.1.0 depends on jaxlib>=0.1.37
chex 0.0.9 depends on jaxlib>=0.1.37
chex 0.0.8 depends on jaxlib>=0.1.37
chex 0.0.7 depends on jaxlib>=0.1.37
chex 0.0.6 depends on jaxlib>=0.1.37
chex 0.0.5 depends on jaxlib>=0.1.37
chex 0.0.4 depends on jaxlib>=0.1.37
chex 0.0.3 depends on jaxlib>=0.1.37
chex 0.0.2 depends on jaxlib>=0.1.37
chex 0.0.1 depends on jaxlib>=0.1.37
To fix this you could try to:
1. loosen the range of package versions you've specified
2. remove package versions to allow pip attempt to solve the dependency conflict
ERROR: ResolutionImpossible: for help visit https://pip.pypa.io/en/latest/topics/dependency-resolution/#dealing-with-dependency-conflicts
conda list output:
# Name Version Build Channel
absl-py 1.4.0 pypi_0 pypi
appdirs 1.4.4 pyh9f0ad1d_0 conda-forge
brotli 1.0.9 hcfcfb64_8 conda-forge
brotli-bin 1.0.9 hcfcfb64_8 conda-forge
brotlipy 0.7.0 py38h91455d4_1005 conda-forge
bzip2 1.0.8 h8ffe710_4 conda-forge
ca-certificates 2023.01.10 haa95532_0
certifi 2022.12.7 py38haa95532_0
cffi 1.15.1 py38h57701bc_3 conda-forge
charset-normalizer 2.1.1 pyhd8ed1ab_0 conda-forge
cloudpickle 2.2.1 pyhd8ed1ab_0 conda-forge
colorama 0.4.6 pyhd8ed1ab_0 conda-forge
contourpy 1.0.7 py38hb1fd069_0 conda-forge
cryptography 39.0.1 py38h95f5157_0 conda-forge
cuda-cccl 12.0.140 0 nvidia
cuda-command-line-tools 12.0.1 0 nvidia
cuda-compiler 12.0.1 0 nvidia
cuda-cudart 12.0.146 0 nvidia
cuda-cudart-dev 12.0.146 0 nvidia
cuda-cuobjdump 12.0.140 0 nvidia
cuda-cupti 12.0.146 0 nvidia
cuda-cuxxfilt 12.0.140 0 nvidia
cuda-documentation 12.0.140 0 nvidia
cuda-libraries 12.0.1 0 nvidia
cuda-libraries-dev 12.0.1 0 nvidia
cuda-nsight-compute 12.0.1 0 nvidia
cuda-nvcc 12.0.140 0 nvidia
cuda-nvdisasm 12.0.140 0 nvidia
cuda-nvml-dev 12.0.140 0 nvidia
cuda-nvprof 12.0.146 0 nvidia
cuda-nvprune 12.0.140 0 nvidia
cuda-nvrtc 12.0.140 0 nvidia
cuda-nvrtc-dev 12.0.140 0 nvidia
cuda-nvtx 12.0.140 0 nvidia
cuda-nvvp 12.0.146 0 nvidia
cuda-opencl 12.0.140 0 nvidia
cuda-opencl-dev 12.0.140 0 nvidia
cuda-profiler-api 12.0.140 0 nvidia
cuda-python 12.0.0 py38h9a6739d_0 nvidia
cuda-sanitizer-api 12.0.140 0 nvidia
cuda-toolkit 12.0.1 0 nvidia
cuda-tools 12.0.1 0 nvidia
cuda-visual-tools 12.0.1 0 nvidia
cudatoolkit 11.3.1 h59b6b97_2
cudnn 8.2.1 cuda11.3_0
cycler 0.11.0 pyhd8ed1ab_0 conda-forge
cython 0.29.33 py38hd3f51b4_0 conda-forge
fonttools 4.38.0 py38h91455d4_1 conda-forge
freetype 2.12.1 h546665d_1 conda-forge
gettext 0.21.1 h5728263_0 conda-forge
glib 2.74.1 h12be248_1 conda-forge
glib-tools 2.74.1 h12be248_1 conda-forge
gst-plugins-base 1.22.0 h001b923_0 conda-forge
gstreamer 1.22.0 h6b5321d_0 conda-forge
gym 0.21.0 py38h595d716_2 conda-forge
gymnasium 0.27.1 pypi_0 pypi
gymnasium-notices 0.0.1 pypi_0 pypi
icu 70.1 h0e60522_0 conda-forge
idna 3.4 pyhd8ed1ab_0 conda-forge
importlib-metadata 4.13.0 pypi_0 pypi
importlib-resources 5.12.0 pyhd8ed1ab_0 conda-forge
importlib_resources 5.12.0 pyhd8ed1ab_0 conda-forge
intel-openmp 2023.0.0 h57928b3_25922 conda-forge
jax 0.2.22 pypi_0 pypi
jax-jumpy 0.2.0 pypi_0 pypi
jpeg 9e hcfcfb64_3 conda-forge
kiwisolver 1.4.4 py38hb1fd069_1 conda-forge
krb5 1.20.1 heb0366b_0 conda-forge
lcms2 2.14 ha5c8aab_1 conda-forge
lerc 4.0.0 h63175ca_0 conda-forge
libblas 3.9.0 16_win64_mkl conda-forge
libbrotlicommon 1.0.9 hcfcfb64_8 conda-forge
libbrotlidec 1.0.9 hcfcfb64_8 conda-forge
libbrotlienc 1.0.9 hcfcfb64_8 conda-forge
libcblas 3.9.0 16_win64_mkl conda-forge
libclang 15.0.7 default_h77d9078_1 conda-forge
libclang13 15.0.7 default_h77d9078_1 conda-forge
libcublas 12.0.2.224 0 nvidia
libcublas-dev 12.0.2.224 0 nvidia
libcufft 11.0.1.95 0 nvidia
libcufft-dev 11.0.1.95 0 nvidia
libcurand 10.3.1.124 0 nvidia
libcurand-dev 10.3.1.124 0 nvidia
libcusolver 11.4.3.1 0 nvidia
libcusolver-dev 11.4.3.1 0 nvidia
libcusparse 12.0.1.140 0 nvidia
libcusparse-dev 12.0.1.140 0 nvidia
libdeflate 1.17 hcfcfb64_0 conda-forge
libffi 3.4.2 h8ffe710_5 conda-forge
libglib 2.74.1 he8f3873_1 conda-forge
libhwloc 2.8.0 h039e092_1 conda-forge
libiconv 1.17 h8ffe710_0 conda-forge
liblapack 3.9.0 16_win64_mkl conda-forge
libnpp 12.0.1.104 0 nvidia
libnpp-dev 12.0.1.104 0 nvidia
libnvjitlink 12.0.140 0 nvidia
libnvjitlink-dev 12.0.140 0 nvidia
libnvjpeg 12.0.1.102 0 nvidia
libnvjpeg-dev 12.0.1.102 0 nvidia
libnvvm-samples 12.0.140 0 nvidia
libogg 1.3.4 h8ffe710_1 conda-forge
libpng 1.6.39 h19919ed_0 conda-forge
libsqlite 3.40.0 hcfcfb64_0 conda-forge
libtiff 4.5.0 hf8721a0_2 conda-forge
libvorbis 1.3.7 h0e60522_0 conda-forge
libwebp-base 1.2.4 h8ffe710_0 conda-forge
libxcb 1.13 hcd874cb_1004 conda-forge
libxml2 2.10.3 hc3477c8_0 conda-forge
libzlib 1.2.13 hcfcfb64_4 conda-forge
luxai-s2 2.1.8 pypi_0 pypi
m2w64-gcc-libgfortran 5.3.0 6 conda-forge
m2w64-gcc-libs 5.3.0 7 conda-forge
m2w64-gcc-libs-core 5.3.0 7 conda-forge
m2w64-gmp 6.1.0 2 conda-forge
m2w64-libwinpthread-git 5.0.0.4634.697f757 2 conda-forge
matplotlib 3.7.0 py38haa244fe_0 conda-forge
matplotlib-base 3.7.0 py38h528a6c7_0 conda-forge
mkl 2022.1.0 h6a75c08_874 conda-forge
msys2-conda-epoch 20160418 1 conda-forge
munkres 1.1.4 pyh9f0ad1d_0 conda-forge
nsight-compute 2022.4.1.6 0 nvidia
numpy 1.24.2 py38h7ec9225_0 conda-forge
openjpeg 2.5.0 ha2aaf27_2 conda-forge
openssl 3.0.8 hcfcfb64_0 conda-forge
opt-einsum 3.3.0 pypi_0 pypi
packaging 23.0 pyhd8ed1ab_0 conda-forge
pcre2 10.40 h17e33f8_0 conda-forge
pettingzoo 1.22.3 pypi_0 pypi
pillow 9.4.0 py38h087119c_1 conda-forge
pip 23.0.1 pyhd8ed1ab_0 conda-forge
ply 3.11 py_1 conda-forge
pooch 1.6.0 pyhd8ed1ab_0 conda-forge
pthread-stubs 0.4 hcd874cb_1001 conda-forge
pthreads-win32 2.9.1 hfa6e2cd_3 conda-forge
pycparser 2.21 pyhd8ed1ab_0 conda-forge
pygame 2.1.3 pypi_0 pypi
pyopenssl 23.0.0 pyhd8ed1ab_0 conda-forge
pyparsing 3.0.9 pyhd8ed1ab_0 conda-forge
pyqt 5.15.7 py38hd6c051e_3 conda-forge
pyqt5-sip 12.11.0 py38hd3f51b4_3 conda-forge
pysocks 1.7.1 pyh0701188_6 conda-forge
python 3.8.16 h4de0772_1_cpython conda-forge
python-dateutil 2.8.2 pyhd8ed1ab_0 conda-forge
python_abi 3.8 3_cp38 conda-forge
pywin32 305 py38h2bbff1b_0
qt-main 5.15.8 h720456b_6 conda-forge
requests 2.28.2 pyhd8ed1ab_0 conda-forge
scipy 1.10.0 py38h0f6ee2a_2 conda-forge
setuptools 56.0.0 pypi_0 pypi
sip 6.7.7 py38hd3f51b4_0 conda-forge
six 1.16.0 pyh6c4a22f_0 conda-forge
tbb 2021.7.0 h91493d7_1 conda-forge
termcolor 2.2.0 pypi_0 pypi
tk 8.6.12 h8ffe710_0 conda-forge
toml 0.10.2 pyhd8ed1ab_0 conda-forge
tornado 6.2 py38h91455d4_1 conda-forge
tqdm 4.64.1 pyhd8ed1ab_0 conda-forge
typing-extensions 4.5.0 pypi_0 pypi
ucrt 10.0.22621.0 h57928b3_0 conda-forge
unicodedata2 15.0.0 py38h91455d4_0 conda-forge
urllib3 1.26.14 pyhd8ed1ab_0 conda-forge
vc 14.3 hb6edc58_10 conda-forge
vec-noise 1.1.4 pypi_0 pypi
vs2015_runtime 14.34.31931 h4c5c07a_10 conda-forge
wheel 0.38.4 pyhd8ed1ab_0 conda-forge
win_inet_pton 1.1.0 pyhd8ed1ab_6 conda-forge
wincertstore 0.2 py38haa95532_2
xorg-libxau 1.0.9 hcd874cb_0 conda-forge
xorg-libxdmcp 1.1.3 hcd874cb_0 conda-forge
xz 5.2.6 h8d14728_0 conda-forge
yaml 0.2.5 h8ffe710_2 conda-forge
zipp 3.14.0 pyhd8ed1ab_0 conda-forge
zstd 1.5.2 h12be248_6 conda-forge
pip list output:
Package Version
------------------- ---------
absl-py 1.4.0
appdirs 1.4.4
brotlipy 0.7.0
certifi 2022.12.7
cffi 1.15.1
charset-normalizer 2.1.1
cloudpickle 2.2.1
colorama 0.4.6
contourpy 1.0.7
cryptography 39.0.1
cuda-python 12.0.0
cycler 0.11.0
Cython 0.29.33
fonttools 4.38.0
gym 0.21.0
gymnasium 0.27.1
gymnasium-notices 0.0.1
idna 3.4
importlib-metadata 4.13.0
importlib-resources 5.12.0
jax 0.2.22
jax-jumpy 0.2.0
kiwisolver 1.4.4
luxai-s2 2.1.8
matplotlib 3.7.0
munkres 1.1.4
numpy 1.24.2
opt-einsum 3.3.0
packaging 23.0
PettingZoo 1.22.3
Pillow 9.4.0
pip 23.0.1
ply 3.11
pooch 1.6.0
pycparser 2.21
pygame 2.1.3
pyOpenSSL 23.0.0
pyparsing 3.0.9
PyQt5 5.15.7
PyQt5-sip 12.11.0
PySocks 1.7.1
python-dateutil 2.8.2
pywin32 305.1
requests 2.28.2
scipy 1.10.0
setuptools 65.6.3
sip 6.7.7
six 1.16.0
termcolor 2.2.0
toml 0.10.2
tornado 6.2
tqdm 4.64.1
typing_extensions 4.5.0
unicodedata2 15.0.0
urllib3 1.26.14
vec-noise 1.1.4
wheel 0.38.4
win-inet-pton 1.1.0
wincertstore 0.2
zipp 3.14.0
StoneT2000 commented
Hi, unfortunately jax doesn’t have great support for windows out of the box, seeing you can try using https://github.com/cloudhan/jax-windows-builder to get it working.
Nearsightedgiraffe commented
I finally got it all working on windows this morning. I found the easiest way was to install jaxlib manually using the wheel version that matches your version from the list here: https://whls.blob.core.windows.net/unstable/index.html
From there, I was able to get the rest working
cs20131516 commented
I finally got it all working on windows this morning. I found the easiest way was to install jaxlib manually using the wheel version that matches your version from the list here: https://whls.blob.core.windows.net/unstable/index.html
From there, I was able to get the rest working
It works wells when i download it. But some errors are occurs on train.py code. Did you solve it?