Files
dockerfile_dp/ldh/.deepspeed_env
2024-08-28 15:18:15 +08:00

7 lines
432 B
Plaintext

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