diff --git a/finetune/docker-compose_pytorch2.3.yml b/finetune/docker-compose_pytorch2.3.yml index 10b5e7a..09db6ed 100644 --- a/finetune/docker-compose_pytorch2.3.yml +++ b/finetune/docker-compose_pytorch2.3.yml @@ -34,8 +34,9 @@ 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" + HTTP_PROXY: "socks5://127.0.0.1:40000" + HTTPS_PROXY: "socks5://127.0.0.1:40000" + BUILDKIT_SYNTAX: "docker/dockerfile:1" volumes: - ./src:/bbtft container_name: ubuntu-finetune