mx.context.num_gpus() not working as expected
AmitPress opened this issue · 2 comments
Description
In kaggle, I turned on the T4 x2, but the function can't detect the gpu.
# setting up context
# if mx.context.num_gpus()>0:
# ctx = mx.gpu()
# else:
# ctx = mx.cpu()
# print(ctx)
mx.context.num_gpus()gives "0" as result
Welcome to Apache MXNet (incubating)! We are on a mission to democratize AI, and we are glad that you are contributing to it by opening this issue.
Please make sure to include all the relevant context, and one of the @apache/mxnet-committers will be here shortly.
If you are interested in contributing to our project, let us know! Also, be sure to check out our guide on contributing to MXNet and our development guides wiki.
Solved: It was a cumbersome process but find out they release specific gpu packages such as mxnet-cu117 etc for gpu support. Still the process is very hazy as you will still need mxnet package and the versions need to be same.