首先检查cuda版本
NVDIA命令:
nvcc –version
如果你装有pytorch,那么可以用
torch.version.cuda
然后安装对应版本的mxnet
pip install + 包名 即可