适配CUDA 12.0的MXNet版本及安装方法咨询

兼容的MXNet版本 #

MXNet从1.9.0版本开始正式支持CUDA 12.0，推荐选用1.9.0及以上的稳定版本（如1.10.0、1.11.0），这些版本对A100的安培架构也有针对性优化。

安装方法 #

1. 预编译包安装（推荐） #

直接通过pip安装适配CUDA 12.0的MXNet预编译包：

• 安装最新兼容版本：

  pip install mxnet-cu120
  

• 指定具体版本（比如1.10.0）：

  pip install mxnet-cu120==1.10.0
  

2. 验证安装有效性 #

安装完成后，运行以下Python代码确认GPU支持正常：

import mxnet as mx
# 输出可用GPU数量，应显示4（对应你的4张A100）
print(mx.context.num_gpus())
# 测试在GPU上创建数组，无报错则说明CUDA支持正常
mx.nd.ones((2, 3), ctx=mx.gpu(0))

3. 关键注意事项 #

• 确保云平台已安装CUDA 12.0开发工具包（nvidia-smi显示的是驱动兼容的CUDA版本，需额外安装cuda-toolkit-12.0才能支持MXNet编译/调用）
• 若使用虚拟环境，需确保在目标环境内执行安装命令
• 避免同时安装CPU版MXNet（mxnet包）和GPU版，否则会导致冲突

内容的提问来源于stack exchange，提问作者Aleksandar