Theano 简介及其安装

简介

Theano 是一个 Python 科学计算库,允许我们进行符号运算,并在 CPUGPU 上执行。

它最初由 Montreal 大学的机器学习研究者们所开发,用来进行机器学习的计算。

按照官网上的说明,它拥有以下几个方面的特点:

  • Numpy, Scipy 的紧密结合
  • GPU 加速
  • 高效的符号计算
  • 速度和稳定性
  • 动态生成 C 代码

使用 anaconda 安装 theano

windows 下,使用 anaconda 安装 theano 的命令为:

  1. conda install mingw libpython
  2. pip install theano

linux 下,使用 anaconda 安装的命令为

  1. conda install theano

安装好之后,还需要安装 Cuda 并进行 GPU 环境的配置,否则是不能利用 GPU 进行计算的,推荐使用 linux/mac 进行配置,具体方法可以参考官网上的配置说明。

查看安装的版本:

In [1]:

  1. import theano
  2.  
  3. theano.__version__

Out[1]:

  1. '0.7.0.dev-54186290a97186b9c6b76317e007844529a352f4'

查看当前使用的 device:

In [2]:

  1. theano.config.device

Out[2]:

  1. 'cpu'

运行测试:

In [3]:

  1. theano.test()
  1. /usr/local/lib/python2.7/dist-packages/theano/misc/pycuda_init.py:34: UserWarning: PyCUDA import failed in theano.misc.pycuda_init
  2. warnings.warn("PyCUDA import failed in theano.misc.pycuda_init")
  3. ....................S...............
  1. Theano version 0.7.0.dev-54186290a97186b9c6b76317e007844529a352f4
  2. theano is installed in /usr/local/lib/python2.7/dist-packages/theano
  3. NumPy version 1.10.1
  4. NumPy relaxed strides checking option: True
  5. NumPy is installed in /usr/lib/python2.7/dist-packages/numpy
  6. Python version 2.7.6 (default, Jun 22 2015, 17:58:13) [GCC 4.8.2]
  7. nose version 1.3.7
  1. ----------------------------------------------------------------------
  2. Ran 37 tests in 37.919s
  3.  
  4. OK (SKIP=1)

Out[3]:

  1. <nose.result.TextTestResult run=37 errors=0 failures=0>

这里我已经在本地 Windows 配好了 GPU 的设置,如果没有配好,显示的结果可能不一样。

Windows 下第一次运行可能会显示 DEBUG: nvcc STDOUT 等内容,Just ignore it!

原文: https://nbviewer.jupyter.org/github/lijin-THU/notes-python/blob/master/09-theano/09.01-introduction-and-installation.ipynb