您的位置:首页 > 其它

Windows下MXnet GPU版本环境搭建

2017-10-30 09:24 211 查看
本系统是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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐