Windows10使用pip安装MXNet GPU版本的那些坑
安装MXNet GPU版本的前提是已经根据官方要求安装好了CUDA和cudnn。
本机具体环境:
Windows10
CUDA version:CUDA 10.1
cudnn version:cudnn 7.6.5
python:3.6
官网指南:
官网链接:http://mxnet.incubator.apache.org/get_started?
结果发现MXNet 1.6 在window下还未支持pip安装
只好按照下面的配置表:选择了 MXnet 1.5 的版本(即mxnet-cu101)
MXNet默认pip指令:
pip install mxnet-cu101
这里默认是直