qiao-lang1997
qiao-lang1997
Seems to be a download failure. (base) vipuser@ubuntu1804:~/project/CLIP_prefix_caption$ sudo cog --debug predict -i [email protected] ⚠ Cog doesn't know if CUDA 10.2 is compatible with PyTorch 1.9.1. This might cause CUDA...
cog.yaml image: r8.im/rmokady/clip_prefix_caption build: gpu: true cuda: "10.2" system_packages: - "libgl1-mesa-glx" - "libglib2.0-0" python_version: "3.8" python_packages: - "transformers==4.11.3" - "git+https://github.com/openai/CLIP.git" - "torch==1.9.1" - "numpy==1.19.5" - "pillow==8.3.2" - "scikit-image==0.16.2" predict: "predict.py:Predictor"