Compile with USE_CUDA=1 to enable GPU usage

MXNet Gluon
#1

以下的讨论是基于:
MXNet版本: 可运行这个来得到pip show mxnet; python -c 'import mxnet; print mxnet.__version__'
操作系统:

图片1
图片11316×233 176 KB

我的版本CUDA是 CUDA Version: 11.8
而 mxnet=cu117
是不是这个原因 要怎么修改

1赞
#2

GPU support is disabled. Compile MXNet with USE_CUDA=1 to enable GPU support.
我也是这样，我的cuda版本是11.6，然后把cpu版本的mxnet卸载了，安装的mxnet-cu116，但是我导入mxnet库显示库不存在。不能使用gpu
是不是cuda版本不能大于10.1啊，我看网上都是10.1的版本
最多mxnet-cu101