configure env

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.

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