MXnet for GPU Install

就算是官网的教程也不要轻易相信=-=

安装的时候安装教程是这样的
(截图留证)

安装教程

好的, 那我就先去安装了个全裸的VS下来.

第二步推荐安装9.2或者9.0版本的, 好啊, 官网推荐啥我就装啥

结果去下载页一看…
download

一个版本9的都没有????
Linux和Mac的也看过了,更没有版本9.

只好去Bing一下教程, 查到的都是一些比较旧的, 说到8.0版本的
(可是也没有8.0版本的下载啊!!!!)

8.1应该差不了多少吧,于是装了8.1的.
不过点击download的时候感到一丝疑惑,文件名是cuda_10.0.130_411.31_windows.exe, 看起来是版本10.0的样子,而且点去10那一看文件名一毛一样…

行吧, 还是先装了8.1

安装好后检查版本

1
$ nvcc -V 
2
nvcc: NVIDIA (R) Cuda compiler driver
3
Copyright (c) 2005-2018 NVIDIA Corporation
4
Built on Sat_Aug_25_21:08:04_Central_Daylight_Time_2018
5
Cuda compilation tools, release 10.0, V10.0.130

好嘛, 哪里是8.1的样子, 分明就是个10.0

行吧, 教程是信不过的了, 直接跳步吧, 那个NVIDIA_CUDA_DNN点过去一看也是无敌复杂, 心累

根据版本10.0安装

1
pip install mxnet-cu100

打开Python检测

>>> import mxnet as mx
>>> from mxnet import nd

>>> nd.array([1,2,3])
[1. 2. 3.]
<NDArray 3 @cpu(0)>

>>> nd.array([1,2,3],ctx=mx.gpu(0))
[1. 2. 3.]
<NDArray 3 @gpu(0)>

搞定, 收工.

(其实还走过很多弯路, 不在赘述了, 大家能快速安装好就好了)

下次再也不相信官网的过时教程了