mm-fit
mm-fit copied to clipboard
[just FYI] Updated installation guide
Hi David and team --
I'm reproducing your results, but I hit a couple of issues in installation. Specifically, some of the versions of python packages that are listed in environment.yml are no longer available. However, the following approach to installation worked for me on an Ubuntu machine with an NVIDIA GPU. (I am sure these packages will also be outdated someday, but it works for now.)
This is my requirements.txt:
pandas
tqdm
numpy>=1.22,<1.27
pandas>=1.3,<3.0
jupyterlab
scipy>=1.7,<1.13
--extra-index-url https://download.pytorch.org/whl/cu126
torch
torchvision
torchaudio
Now, install it in conda:
conda create -n mm-fit python==3.9
pip install -r requirements.txt
Done! Now I can train the model.