File size: 550 Bytes
a00ee36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
build:
gpu: true
python_version: "3.8"
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
python_packages:
- "torch==1.8.0"
- "torchvision==0.9.0"
- "numpy==1.21.1"
- "ipython==7.21.0"
- "pytorch-pretrained-biggan==0.1.1"
- "ftfy==6.0.3"
- "cma==3.1.0"
- "scikit-learn==0.24.2"
- "imageio==2.9.0"
- "Pillow==8.3.1"
- "regex==2021.8.28"
- "nltk==3.6.3"
- "scipy==1.7.1"
- "h5py==3.4.0"
- "matplotlib==3.4.3"
- "faiss-gpu==1.7.1.post2"
predict: "predict.py:Predictor"
|