diff --git a/README.md b/README.md index 87face3..9e4b9ae 100644 --- a/README.md +++ b/README.md @@ -37,7 +37,11 @@ The repo is still being under construction, thanks for your patience. ```angular2html conda create -n unique3d conda activate unique3d +# Match your cuda version. +pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cu121 pip install -r requirements.txt +# Install xformers, this will find proper version, but not install torch again. +pip install xformers torch==2.1.0+cu121 ``` ### Windows Setup. diff --git a/requirements-detail.txt b/requirements-detail.txt deleted file mode 100644 index cd0bc9a..0000000 --- a/requirements-detail.txt +++ /dev/null @@ -1,27 +0,0 @@ -accelerate==0.29.2 -datasets==2.18.0 -diffusers==0.27.2 -fire==0.6.0 -gradio==4.32.0 -jaxtyping==0.2.29 -numba==0.59.1 -numpy==1.26.4 -nvdiffrast==0.3.1 -omegaconf==2.3.0 -onnxruntime_gpu==1.17.0 -opencv_python==4.9.0.80 -opencv_python_headless==4.9.0.80 -ort_nightly_gpu==1.17.0.dev20240118002 -peft==0.10.0 -Pillow==10.3.0 -pygltflib==1.16.2 -pymeshlab==2023.12.post1 -pytorch3d==0.7.5 -rembg==2.0.56 -torch==2.1.0+cu121 -torch_scatter==2.1.2 -tqdm==4.64.1 -transformers==4.39.3 -trimesh==4.3.0 -typeguard==2.13.3 -wandb==0.16.6 diff --git a/requirements.txt b/requirements.txt index e0eba37..d82420a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,28 +1,26 @@ -accelerate -datasets -diffusers>=0.26.3 -fire -gradio -jaxtyping -numba -numpy +accelerate==0.29.2 +datasets==2.18.0 +diffusers==0.27.2 +fire==0.6.0 +gradio==4.32.0 +jaxtyping==0.2.29 +numba==0.59.1 +numpy==1.26.4 git+https://github.com/NVlabs/nvdiffrast.git -omegaconf>=2.3.0 -onnxruntime_gpu -opencv_python -opencv_python_headless +omegaconf==2.3.0 +onnxruntime_gpu==1.17.0 +opencv_python==4.9.0.80 +opencv_python_headless==4.9.0.80 ort_nightly_gpu -peft -Pillow -pygltflib -pymeshlab>=2023.12 +peft==0.10.0 +Pillow==10.3.0 +pygltflib==1.16.2 +pymeshlab==2023.12.post1 git+https://github.com/facebookresearch/pytorch3d.git@stable -rembg[gpu] -torch>=2.0.1 -torch_scatter -tqdm -transformers -trimesh -typeguard -wandb -xformers +rembg==2.0.56 +torch_scatter==2.1.2 +tqdm==4.64.1 +transformers==4.39.3 +trimesh==4.3.0 +typeguard==2.13.3 +wandb==0.16.6