search

mir-aidj / all-in-one

All-In-One Music Structure Analyzer
http://arxiv.org/abs/2307.16425
MIT License
370 stars 35 forks source link

Added dockerfile, updated README #6

Open Kaszanas opened 8 months ago

Kaszanas commented 8 months ago

As the installation instructions are quite hefty, I have decided to add a Dockerfile. This streamlines the environment setup for anyone using containers.

This was especially helpful for me because installation on Windows failed for me (I am not sure why).

Also I was not sure if you use the GPU capabilities but the image that is built uses PyTorch with CUDA.

There is an alternative to this that may be worth exploring: https://hub.docker.com/r/pytorch/pytorch/tags

This could remove one of the RUN commands:

RUN pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118