Explorar el Código

Fix pip requirements issues

5.0
Adam hace 4 años
padre
commit
faa6aeaa20
Se han modificado 1 ficheros con 3 adiciones y 1 borrados
  1. +3
    -1
      requirements.txt

+ 3
- 1
requirements.txt Ver fichero

@@ -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

Cargando…
Cancelar
Guardar