diff --git a/Dockerfile.cpu b/Dockerfile.cpu index b437e1e..f59efe9 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -10,13 +10,8 @@ RUN pip3 install \ torchaudio==0.10.2+cpu \ -f https://download.pytorch.org/whl/cpu/torch_stable.html -RUN pip3 install \ - opencv-python \ - sklearn \ - matplotlib \ - pandas \ - tqdm +COPY requirements.txt /requirements.txt -#COPY src /src +RUN pip3 install -r requirements.txt ENTRYPOINT ["bash"] \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..c9b69a2 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,29 @@ +cycler==0.11.0 +fonttools==4.33.2 +imageio==2.17.0 +joblib==1.1.0 +kiwisolver==1.4.2 +matplotlib==3.5.1 +networkx==2.8 +numpy==1.22.3 +opencv-python==4.5.5.64 +packaging==21.3 +pandas==1.4.2 +Pillow==9.0.1 +pyparsing==3.0.8 +python-dateutil==2.8.2 +pytz==2022.1 +PyWavelets==1.3.0 +PyYAML==6.0 +scikit-image==0.19.2 +scikit-learn==1.0.2 +scipy==1.8.0 +six==1.16.0 +sklearn==0.0 +threadpoolctl==3.1.0 +tifffile==2022.4.22 +torch==1.10.2+cpu +torchaudio==0.10.2+cpu +torchvision==0.11.3+cpu +tqdm==4.64.0 +typing_extensions==4.1.1