TuSimple / TuSimple-DUC

Understanding Convolution for Semantic Segmentation
https://arxiv.org/abs/1702.08502
Apache License 2.0
610 stars 118 forks source link

Fail to use other gpus except gpus =0 #25

Closed Serian1992 closed 6 years ago

Serian1992 commented 6 years ago

Hi,

I tried to reset the parameter 'gpus = 3', an error appeared (seen in the fig below). My current environment is Python 2.7, Mxnet version 1.0.0 under Win10 with 4 available GPUs. Any suggestion for this problem? Thanks a lot! b06d7a61ad612b7e3edf2f2678936496

GrassSunFlower commented 6 years ago

Actually what you provide in the figure has nothing to do with the error message but some normal message logged by mxnet. Can you provide the exact error message? Besides, I would recommend you to try run some basic mxnet examples with gpu No.3 to make sure everything is OK with your hardware, driver and MxNet.