You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.
I compiled mxnet with MXNET_USE_CUDA=1, sometimes I didn't use gpu, in this situation it made me to wait almost 2 minutes to watch the iterations running, compare with before I didn't use MXNET_USE_CUDA=1.
Comm() use MXNET_USE_CUDA a compile macro to get the context, which is called from _initialize_kvstore in model.py, finally, it case to active gpu, this made me to wait a long time(8 Tesla M40 crads on board) even i didn't use gpu.
I think, it's better to determine pinned_ctx by a runtime variable like a command argument, rather than a compile macro.
The text was updated successfully, but these errors were encountered:
I compiled mxnet with MXNET_USE_CUDA=1, sometimes I didn't use gpu, in this situation it made me to wait almost 2 minutes to watch the iterations running, compare with before I didn't use MXNET_USE_CUDA=1.
debuged around, found in
src/kvstore/comm.h
Comm() use MXNET_USE_CUDA a compile macro to get the context, which is called from
_initialize_kvstore
inmodel.py
, finally, it case to active gpu, this made me to wait a long time(8 Tesla M40 crads on board) even i didn't use gpu.I think, it's better to determine
pinned_ctx
by a runtime variable like a command argument, rather than a compile macro.The text was updated successfully, but these errors were encountered: