lingyuzeng/cdc_dockerfile
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
hotwa
2024-06-21 15:12:44 +08:00
parent 5c6e2f44e5
commit be77975ce2
7 changed files with 20 additions and 900 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/docker-compose_pytorch1.13.yml
View File

@@ -19,7 +19,7 @@ services:
USE_ROCM: 0
USE_XPU: 0
CUDA: cu117
CUDA_ARCH_LIST: "80;86" # for RTX 4090, all : "80;86;89;90"
CUDA_ARCH_LIST: "80;86" # for RTX 4090, all : "80;86;89;90" 编译deepspeed内核需要，这个参数很严格
SETUPTOOLS_VERSION: "69.5.1"
ROOT_PASSWD: "root"
DCUTLASS_NVCC_ARCHS: "90a" # 90a for H100 ,89:GeForce RTX 4090