Windows下MXnet GPU版本环境搭建

来源:互联网 发布:js涂膜防水施工工艺 编辑:程序博客网 时间:2024/06/09 18:13

本系统是windows10系统,不同系统下载相应的cuda版本即可。

1.下载cuda(CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题)
Cuda 8.0 window10版本下载地址: https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda_8.0.44_win10-exe

直接像安装其他应用程序一样安装cuda。

2.测试是否安装成功:
检查nvcc编译器是否正确安装,在命令提示符窗口中输入:nvcc -V
回车查看是否有版本信息。若出现版本信息,则证明nvcc安装成功。

3.然后卸载cpu版的mxnet:pip uninstall mxnet
下载GPU版的mxnet:pip install –pre mxnet-cu80

原创粉丝点击