Skip to content

Commit fd1b0f7

Browse files
committed
Better unknown GPU message
1 parent d1be961 commit fd1b0f7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

neuralstyle/algorithms.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,7 @@ def gpuname():
354354
try:
355355
gpus = GPUtil.getGPUs()
356356
except:
357-
LOGGER.warning("Unable to detect GPU model")
357+
LOGGER.warning("Unable to detect GPU model. Is your GPU configured? Are you running with nvidia-docker?")
358358
return "UNKNOWN"
359359
if len(gpus) == 0:
360360
raise ValueError("No GPUs detected in the system")

0 commit comments

Comments
 (0)