Aryeh Rotberg
Updated PyTorch to version 2.3.1+cu121 and TorchVision to 0.18.1+cu121 due to security reasons.
b365d37
| --find-links https://download.pytorch.org/whl/torch_stable.html | |
| fastapi | |
| uvicorn[standard] | |
| matplotlib | |
| numpy | |
| pandas | |
| Pillow | |
| scikit_learn | |
| setuptools | |
| starlette | |
| torch==2.3.1+cu121 | |
| torchvision==0.18.1+cu121 | |
| tqdm | |
| python-multipart | |
| streamlit | |
| mlflow | |
| segmentation-models-pytorch | |
| opencv-python-headless | |