From 0459c9474d034b191c4d60163acc18c8c4446ba3 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Fri, 12 Jul 2024 11:21:41 +0800
Subject: [PATCH] add proxy port

---
 finetune/Dockerfile                    | 4 ++++
 finetune/docker-compose_pytorch2.3.yml | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/finetune/Dockerfile b/finetune/Dockerfile
index d3fb917..6c9ad48 100644
--- a/finetune/Dockerfile
+++ b/finetune/Dockerfile
@@ -6,6 +6,10 @@
 # - https://docs.docker.com/build/dockerfile/frontend/#stable-channel
 ARG CUDA_VERSION=12.1.0
 FROM nvidia/cuda:${CUDA_VERSION}-cudnn8-devel-ubuntu22.04
+ARG HTTP_PROXY
+ARG HTTPS_PROXY
+ENV http_proxy=${HTTP_PROXY}
+ENV https_proxy=${HTTPS_PROXY}
 ARG DEBIAN_FRONTEND="noninteractive"
 ENV DEBIAN_FRONTEND=${DEBIAN_FRONTEND}
 ENV MAMBA_ROOT_PREFIX=~/micromamba
diff --git a/finetune/docker-compose_pytorch2.3.yml b/finetune/docker-compose_pytorch2.3.yml
index a6dc8aa..10b5e7a 100644
--- a/finetune/docker-compose_pytorch2.3.yml
+++ b/finetune/docker-compose_pytorch2.3.yml
@@ -34,6 +34,8 @@ services:
         DCUTLASS_NVCC_ARCHS: "80;86;89;90;90a" # 90a for H100 GPU 89:GeForce RTX 4090 
         DEEPSPEED_VERSION: "master"
         DEEPSPEED_INSTALL_FLAGS: "--allow_sudo"
+        HTTP_PROXY: "http://localhost:15777"
+        HTTPS_PROXY: "http://localhost:15777"
     volumes:
       - ./src:/bbtft
     container_name: ubuntu-finetune