run: PYTORCH_NO_CUDA_MEMORY_CACHING=1 \ PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True \ CUDA_LAUNCH_BLOCKING=1 \ FLASK_APP=src/app.py \ PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128 \ flask run --host=0.0.0.0