Fix pip requirements issues
parent
099e6f5ebd
commit
faa6aeaa20
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue