1、查看显卡适用的cuda版本。
cmd终端下输入Nvidia-smi,CUDA版本是11.0。目前并没有发布mxnet-cuda110版本的轮子,建议下载低于11.0版本的CUDA。轮子可查看网址https://dist.mxnet.io/python
2、下载CUDA Toolkit 10.2。
安装网址:https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork
3、自定义安装CUDA Toolkit 10.2。注意自定义安装选项。
4、根据对应的cuda版本,下载mxnet_cuda102轮子,并在cmd下使用pip安装。下载地址:https://dist.mxnet.io/python
5、找到虚拟环境的地址,激活虚拟环境,重设yml条件,更新虚拟变量设置



