This commit is contained in:
IDGallagher
2025-01-11 21:59:57 +08:00
parent bd4f43e70e
commit 1c574111f5
+2 -12
View File
@@ -15,14 +15,9 @@ conda create -n mvs python=3.9
conda activate mvs
```
Install the [PyTorch and TorchVision](https://pytorch.org/get-started/locally/) versions which are compatible with your CUDA configuration.
Install the [PyTorch and TorchVision](https://pytorch.org/get-started/locally/) versions that are compatible with your CUDA configuration.
```
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
```
[Optional] Install a version of xformers that works with your version of PyTorch
```
pip install xformers --index-url https://download.pytorch.org/whl/cu118
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
```
Install ffmpeg 6.x
@@ -40,11 +35,6 @@ Install faiss https://github.com/facebookresearch/faiss
conda install -c pytorch faiss-cpu=1.9.0
```
[Optional] Set up custom modules from [PyTorch3D](https://github.com/facebookresearch/pytorch3d) to increase speed and reduce memory consumption of interpolation operations.
```
cd dot/utils/torch3d/ && python setup.py install && cd ../../..
```
[Optional] Download watermark removal repo
```
git clone https://github.com/l-comm/WatermarkRemoval.git