8 lines
218 B
Makefile
8 lines
218 B
Makefile
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
|