Skip to content

Commit 7c3725d

Browse files
committed
remove device unsetting from MPS script
1 parent 60ebf00 commit 7c3725d

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

jobqueues/templates/SLURM_job.sh.j2

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -65,11 +65,6 @@ nvidia-cuda-mps-control -d
6565

6666
# start the server
6767
echo "start_server -uid ${UID}" | nvidia-cuda-mps-control
68-
69-
# mps-control remaps the index of the device it is running on to 0 so we
70-
# need to unset it to avoid "cuda_error_no_device"
71-
72-
unset CUDA_VISIBLE_DEVICES
7368
{% endif %}
7469

7570
{% for rsh in runsh %}

0 commit comments

Comments
 (0)