pip conflict when run `conda env create -f scripts/gpu_environment.yml`

young-geng / EasyLM

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

Apache License 2.0

2.33k stars 247 forks source link

pip conflict when run `conda env create -f scripts/gpu_environment.yml` #92

Closed 0x00-pl closed 7 months ago

0x00-pl commented 11 months ago

The conflict is caused by: jax[cuda11-pip] 0.4.14 depends on nvidia-cuda-runtime-cu11>=11.8; extra == "cuda11_pip" torch 2.0.1 depends on nvidia-cuda-runtime-cu11==11.7.99; platform_system == "Linux" and platform_machine == "x86_64"