Spaces:
No application file
No application file
File size: 382 Bytes
81685e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
build:
gpu: true
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
- "python3-setuptools"
python_version: "3.10"
cuda: "11.5" # Query nvcc --version
python_packages:
- "torch==2.3.1"
- torchvision
- torchaudio
- "anomalib[full]==1.1.0"
- "openvino==2024.2.0"
- "lightning==2.3.0"
- "scikit-learn==1.5.0"
train: "train.py:train"
|