安装GPU版Mxnet(cu101)

本文介绍了在Win10系统中如何安装GPU版的Mxnet Cu101,包括下载CUDA 10.1、CuDNN以及Mxnet_cu101,详细阐述了配置环境的过程,如验证CUDA安装、降低CUDA版本以及安装CuDNN的步骤。此外,还提供了检查CUDA版本和安装Mxnet的指导。
摘要由CSDN通过智能技术生成

Win10系统

下载CUDA、CuDNN和Mxnet_cu101

官网下载CUDA10.1
CUDA10.1
清华源下载Mxnet_cu101
Mxnet_cu101
官网下载CuDNN
cudnn10.1

配置环境

CUDA

按照installer的步骤进行即可

验证是否安装成功

cd到安装目录 cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite

运行 deviceQuery.exebandwidthTest.exe

出现 Result = PASS 说明安好了

降低CUDA版本

NVIDA系列的程序除了图形驱动程序PhysX系统软件不能卸载

其他都卸载(NVIDA HD音频驱动程序可以留着不影响)

重新下载低版本CUDA并安装即可

查看CUDA版本

cd到 cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite
运行 deviceQuery.exe

CUDA Driver Version = 11.1, CUDA Runtime Version = 10.1

驱动版本以及运行版本(ps:驱动版本可能要比运行版本高才行)

Mxnet

pip install mxnet_cu101
或者直接安装.whl文件
pip install ***\mxnet_cu101-1.5.0-py2.py3-none-win_amd64.whl
其中***换成.whl文件所在位置路径

部分教程说要卸载原有的Mxnet
pip uninstall mxnet

CuDNN

下载下来的安装包解压到

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

如图:
在这里插入图片描述
安装包里的四个项目解压出来:安装包里的四个项目解压出来

[END]

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值