Open moming133 opened 3 days ago
torch==2.5.0 transformers==4.45.2 datasets==2.20.0 accelerate==0.26.1 jupyter==1.0.0 numpy==1.26.3 pillow==10.2.0 peft==0.13.2 diffusers==0.30.3 timm==0.9.16 opencv-python gradio spaces insightface onnxruntime
Create a python 3.10.13 conda env (you could also use virtualenv)
conda create -n py310 python=3.10.13
conda activate py310
Install pytorch with your CUDA version, e.g.
pip install torch==2.5.9+cu118 torchvision --extra-index-url https://download.pytorch.org/whl/cu118
torch==2.5.0 transformers==4.45.2 datasets==2.20.0 accelerate==0.26.1 jupyter==1.0.0 numpy==1.26.3 pillow==10.2.0 peft==0.13.2 diffusers==0.30.3 timm==0.9.16 opencv-python gradio spaces insightface onnxruntime