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

add 4090 arch 8.9

Browse Source
This commit is contained in:
lingyuzeng
2024-07-15 12:03:45 +08:00
parent 0378cb2509
commit 7b1a20d0b8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/Dockerfile.update
View File

@@ -111,7 +111,7 @@ python3 -m pip uninstall -y apex
EOT EOT
# install apex # install apex
ARG TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6;9.0" ARG TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.9,9.0"
ENV TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST} ENV TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST}
RUN <<EOT RUN <<EOT
#!/bin/bash #!/bin/bash