Skip to content

Commit

Permalink
Add some GPU computing stuff.
Browse files Browse the repository at this point in the history
  • Loading branch information
sfremerey committed Jul 13, 2021
1 parent 9bbc1f4 commit 6b3f31c
Show file tree
Hide file tree
Showing 2 changed files with 33 additions and 0 deletions.
5 changes: 5 additions & 0 deletions README_GPU_Computing.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
Execute the following two commands to install the necessary dependencies for GPU computing:

pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

pip3 install -r requirements_gpu.txt
28 changes: 28 additions & 0 deletions requirements_gpu.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
# pip install -r requirements.txt

# base ----------------------------------------
matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow
PyYAML>=5.3.1
scipy>=1.4.1
tqdm>=4.41.0

# logging -------------------------------------
tensorboard>=2.4.1
# wandb

# plotting ------------------------------------
seaborn>=0.11.0
pandas

# export --------------------------------------
# coremltools>=4.1
# onnx>=1.9.0
# scikit-learn==0.19.2 # for coreml quantization

# extras --------------------------------------
# Cython # for pycocotools https://github.com/cocodataset/cocoapi/issues/172
# pycocotools>=2.0 # COCO mAP
thop # FLOPs computation

0 comments on commit 6b3f31c

Please sign in to comment.