Fix pip requirements issues
This commit is contained in:
1 parent
099e6f5ebd
commit
faa6aeaa20
1 file changed
+3
-1
+3
-1
@@ -1,4 +1,5 @@
|
|||||||
# pip install -U -r requirements.txt
|
# pip install -U -r requirements.txt
|
||||||
|
Cython
|
||||||
numpy==1.17
|
numpy==1.17
|
||||||
opencv-python
|
opencv-python
|
||||||
torch >= 1.5
|
torch >= 1.5
|
||||||
@@ -8,7 +9,8 @@ tqdm
|
|||||||
pillow
|
pillow
|
||||||
tensorboard
|
tensorboard
|
||||||
pyyaml
|
pyyaml
|
||||||
|
torchvision
|
||||||
|
scipy
|
||||||
|
|
||||||
# Nvidia Apex (optional) for mixed precision training --------------------------
|
# 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
|
# 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
|
||||||
|
|||||||
Reference in new issue
Block a user