slurm/dlr: fix workaround logic

This commit is contained in:
Starbeamrainbowlabs 2024-11-14 22:28:39 +00:00
parent 17d2d2bcaf
commit 7c4f3d325d
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -22,7 +22,7 @@ command_exists() {
#######################################################
# Fix "Could not load library libcublasLt.so.12. Error: libcublasLt.so.12: cannot open shared object file: No such file or directory" error
if [[ "${SLURM_CLUSTER_NAME}" != "cs-cluster" ]] && [[ -d "${HOME}/cuda" ]]; then
if [[ "${SLURM_CLUSTER_NAME}" == "cs-cluster" ]] && [[ -d "${HOME}/cuda" ]]; then
echo "[slurm_runner] csgpu cluster detected, sourcing extra CUDA setup script" >&2;
#shellcheck source=/dev/null
source "${HOME}/cuda/activate.sh";