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

cuda 11.7 suport CUDA_ARCH_LIST: "80;86"

Browse Source
This commit is contained in:
hotwa
2024-06-21 15:02:18 +08:00
parent d351fe520a
commit 5c6e2f44e5
1 changed files with 1 additions and 1 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_ROCM: 0
USE_XPU: 0 USE_XPU: 0
CUDA: cu117 CUDA: cu117
CUDA_ARCH_LIST: "80;86;89;90" # for RTX 4090, all : "80;86;89;90" CUDA_ARCH_LIST: "80;86" # for RTX 4090, all : "80;86;89;90"
SETUPTOOLS_VERSION: "69.5.1" SETUPTOOLS_VERSION: "69.5.1"
ROOT_PASSWD: "root" ROOT_PASSWD: "root"
DCUTLASS_NVCC_ARCHS: "90a" # 90a for H100 ,89:GeForce RTX 4090 DCUTLASS_NVCC_ARCHS: "90a" # 90a for H100 ,89:GeForce RTX 4090