add proxy port

2024-07-12 11:21:41 +08:00
parent 6c6027d5f5
commit 0459c9474d
2 changed files with 6 additions and 0 deletions
--- 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