revise makefile

falconlee236 · falconlee236 · commit 573e334a8552 · 2024-11-16T00:50:37.000+09:00
diff --git a/Makefile b/Makefile
@@ -13,6 +13,7 @@ train:
 container:
 	@docker run -itd \
 				--name $(CONTAINER_NAME) \
+				--gpus all \
 				-v /tmp/artifect:/home/$(USER_NAME)/artifect \
 				-p 3030:3030 \
 				--restart always \
