diff --git a/README.md b/README.md index 034faa0..d2386a8 100644 --- a/README.md +++ b/README.md @@ -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