《MXNet深度学习实战》—2.3　本地pip安装MXNet

2.3　本地pip安装MXNet

本地通过pip工具安装MXNet的界面如图2-9所示，这里我以安装支持GPU的MXNet 1.3.1为例介绍如何在本地安装MXNet。首先，用户需要准备好2.1节中的显卡驱动、CUDA和cuDNN，然后才可以开始在本地安装MXNet，因此，相比通过Docker镜像使用MXNet，本地安装支持GPU的MXNet需要额外手动安装CUDA和cuDNN。

图2-9　通过pip方式安装支持GPU的MXNet

安装过程中如果有权限限制问题则可以切换到root用户进行安装或者在命令前加sudo。因为我们需要用apt-get命令来安装Python和pip工具，因此首先需要通过以下命令更新apt-get的软件源：

$ sudo apt-get update

接下来需要安装Python3和pip3。Python3是后期会经常用到的编程语言，pip3则是用于Python3的软件安装工具。可以通过apt-get使用下面这个命令来安装，参数-y表示在安装时默认选择同意继续，若不加这个参数的话就会在安装到一半时需要手动敲一个y表示同意继续安装：

$ sudo apt-get install -y python3 python3-pip

安装成功后，可以使用如下命令查看Python版本：

$ python3 --version

Python 3.5.2

因为我们安装的是支持GPU的MXNet版本，所以在安装MXNet之前可以先用以下命令确认下你的CUDA版本：

$ nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2016 NVIDIA Corporation

Built on Tue_Jan_10_13:22:03_CST_2017

Cuda compilation tools, release 8.0, V8.0.61

接下来就可以用pip3命令安装指定CUDA版本的MXNet，因为CUDA版本是8.0，所以这里安装支持CUDA 8.0版本的MXNet，而且默认安装的是最新版本的MXNet：

$ pip3 install mxnet-cu80

如果你需要安装指定版本的MXNet，比如本书采用的1.3.1版本，那么可以使用以下命令来安装：

$ pip3 install mxnet-cu80==1.3.1

安装成功后，就可以在命令行输入python3进入Python环境（如果需要修改默认的Python环境，则可以修改当前账户的.bashrc文件内容，具体参考2.2.5节）：

$ python3

Python 3.5.2 (default, Nov 12 2018, 13:43:14)

[GCC 5.4.0 20160609] on linux

Type "help", "copyright", "credits" or "license" for more information.

>>> 

可以看到在进入Python环境后，命令行的前缀就变成了“>>>”，这就说明你已经成功进入Python环境了，然后你可以导入MXNet并执行一些简单的操作，代码如下：

import mxnet as mx

a = mx.nd.array([1,2,3]).as_in_context(mx.gpu(0))

b = mx.nd.array([4,5,6]).as_in_context(mx.gpu(0))

print(a+b)

输出结果如下：

[5. 7. 9.]

<NDArray 3 @gpu(0)>

如果你要卸载MXNet，可以使用如下命令：

$ pip3 uninstall mxnet-cu80==1.3.1

当然，我相信你会慢慢喜欢上MXNet的，所以这个命令你可能不会用到。至此，本地安装MXNet就介绍完了。