-
Notifications
You must be signed in to change notification settings - Fork 19
Open
Description
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 torch and 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:
-
Edit
attribute_hallucination\editing_tool\semantic_segmentation_pytorch\lib\utils\data\dataloader.py:
Change _update_worker_pids to _set_worker_pids in Line 238 and Line 3 -
Edit
attribute_hallucination\editing_tool\semantic_segmentation_pytorch\lib\nn\parallel\data_parallel.py
Change in Line 16obj.cuda(dev, async=True)toobj.cuda(dev, non_blocking=True)sinceasyncis a reserved keyword in python 3.7
lucidBrot
Metadata
Metadata
Assignees
Labels
No labels