Skip to content

S
seminar-breakout
Commit ac09ed56 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix GPUUtilTracker

parent 0db4e3df
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/callbacks/prof.py
View file @ ac09ed56
...@@ -37,7 +37,7 @@ class GPUUtilizationTracker(Callback): ...@@ -37,7 +37,7 @@ class GPUUtilizationTracker(Callback):
if env is None: if env is None:
logger.warn("[GPUUtilizationTracker] Both devices and CUDA_VISIBLE_DEVICES are None! " logger.warn("[GPUUtilizationTracker] Both devices and CUDA_VISIBLE_DEVICES are None! "
"Will monitor all visible GPUs!") "Will monitor all visible GPUs!")
self._devices = list(range(get_nr_gpu())) self._devices = list(map(str, range(get_nr_gpu())))
else: else:
self._devices = env.split(',') self._devices = env.split(',')
else: else:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment