uav-vts
/

training_sem

Model card Files Files and versions

kai-2054 commited on Sep 15, 2025

Commit

588cc49

·

1 Parent(s): efa3028

update

Files changed (1) hide show

train_job.py +3 -0

train_job.py CHANGED Viewed

@@ -7,6 +7,9 @@ job = run_job(
     command=[
             "bash",
             "-c",
             "apt-get update && apt-get install -y git && "
             "git clone https://huggingface.co/uav-vts/training_sem . && "
             "pwd && ls -al && "

     command=[
             "bash",
             "-c",
+            "export CUDA_HOME=/usr/local/cuda && "
+            "export PATH=$CUDA_HOME/bin:$PATH && "
+            "export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH && "
             "apt-get update && apt-get install -y git && "
             "git clone https://huggingface.co/uav-vts/training_sem . && "
             "pwd && ls -al && "