Skip to content

Commit 509fb38

Browse files
committed
fix: enable cuda not required - variable set by asking for gres=gpu
1 parent 817399e commit 509fb38

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

nipype/pipeline/plugins/slurm.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,6 @@ def __init__(self, **kwargs):
5353
self._template = open(self._template).read()
5454
if 'sbatch_args' in kwargs['plugin_args']:
5555
self._sbatch_args = kwargs['plugin_args']['sbatch_args']
56-
if 'enable_cuda' in kwargs['plugin_args'] and kwargs['plugin_args']['enable_cuda']:
57-
self._template = '\n'.join((self._template, 'CUDA_VISIBLE_DEVICES=0'))
5856
self._pending = {}
5957
super(SLURMPlugin, self).__init__(self._template, **kwargs)
6058

0 commit comments

Comments
 (0)