安装CUDA和MXNet-cu91
第一步,打开NVIDIA控制面板,查看当前显卡驱动能够支持的CUDA版本:
以下表格可以用来参考:
第二步,在这里,选择对应的CUDA下载:
https://developer.nvidia.com/cuda-toolkit-archive
第三步,然后进行推荐的安装。(自动配置好环境变量的)
安装成功的验证方法是在cmd下输入nvcc -V
第四步,pip uninstall mxnet
接着pip install mxnet-cu91
;------------------------因为我的CUDA版本是9.1
最后,在jupyter notebook验证: