hucvl / attribute_hallucination

125 stars 19 forks source link

Add Windows Support Tutorial in README and a good requirements.txt #5

Open filipppp opened 4 years ago

filipppp commented 4 years ago

I got it running on Windows with following editing_tool/requirements.txt:

cupy-cuda101==7.4.0
cycler==0.10.0
fastrlock==0.4
future==0.18.2
imageio==2.8.0
joblib==0.14.1
kiwisolver==1.2.0
matplotlib==3.2.1
numpy==1.18.4
opencv-python==4.2.0.34
Pillow==7.1.2
pynvrtc==9.2
pyparsing==2.4.7
python-dateutil==2.8.1
scikit-learn==0.22.2.post1
scipy==1.2.0
six==1.14.0
sklearn==0.0
torch==1.5.0+cu101
torchvision==0.6.0+cu101
tqdm==4.46.0

You should maybe take out torchand cupy since they are installed for CUDA 10.1. To get it to work on Windows 10 with Python3.7 you have to do the following:

leventkaracan commented 4 years ago

Thanks for your efforts to run our model on Windows machines!