Windows安装mxnet带GPU上下文

---

前期准备

名称	版本	网址
GPU	NVIDIA	-
python2 (anaconda)	64bit ( python2/3均可 )	https://www.anaconda.com/download
CUDA	8.0 / 9.0 / 9.2	https://developer.nvidia.com/cuda-toolkit-archive
CUDNN	8.0 / 9.0 / 9.2	https://developer.nvidia.com/cudnn

mxnet与CUDN的配对看图：

注意：

• python一定要64位的才能用GPU
• CUDA和CUDNN需要版本一致
• 命令nvcc --version可以看CUDA的版本

推荐版本：

• CUDA、CUDNN：9.0
• python：anaconda3

---

安装步骤

1. 安装anaconda

1. 双击安装包
2. 下一步、下一步、下一步
3. 可以添加环境变量，也可以不添加

2. 安装CUDA

1. 双击解压包解压（解压到哪都可以，解压出来的临时文件会自动删除）
2. 安装、下一步、下一步
3. 如果安装visual studio套件失败，就进自定义把visual studio去掉

3. 安装CUDNN

• 将CUDNN中的文件夹bin、include、lib\x64中的文件拷贝的CUDA对应文件夹中

4. 安装mxnet

• pip install mxnet-cu80（CUDN版本为8.0）
• pip install mxnet-cu90（CUDN版本为9.0）
• pip install mxnet-cu92（CUDN版本为9.2）