From faa6aeaa20cc52f3114811e302ca167cfbce7cd7 Mon Sep 17 00:00:00 2001 From: Adam Date: Sun, 14 Jun 2020 18:18:31 +1000 Subject: [PATCH] Fix pip requirements issues --- requirements.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 1328818..fced686 100755 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ # pip install -U -r requirements.txt +Cython numpy==1.17 opencv-python torch >= 1.5 @@ -8,7 +9,8 @@ tqdm pillow tensorboard pyyaml - +torchvision +scipy # Nvidia Apex (optional) for mixed precision training -------------------------- # git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex