This commit is contained in:
Glenn Jocher 2020-01-23 12:24:52 -08:00
parent 52041cffb9
commit ee4f7a324d
1 changed files with 1 additions and 1 deletions

View File

@ -5,7 +5,7 @@ torch >= 1.4
matplotlib matplotlib
pycocotools pycocotools
tqdm tqdm
Pillow pillow
# 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