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

first add

Browse Source
This commit is contained in:
lingyuzeng
2024-08-28 15:18:15 +08:00
commit 873429d4e6
57 changed files with 4892 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ldh/.deepspeed_env Normal file
View File

@@ -0,0 +1,6 @@
CUDA_HOME=/usr/local/cuda
CUTLASS_PATH=/opt/cutlass
TORCH_CUDA_ARCH_LIST="8.0 9.0+PTX"
PATH=/opt/openmpi/bin:/usr/lib/jvm/default-java/bin:/usr/local/cuda/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
LD_LIBRARY_PATH=/opt/openmpi/lib:/usr/local/cuda/lib64:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64/stubs