Theano
是一个 Python
科学计算库,允许我们进行符号运算,并在 CPU
和 GPU
上执行。
它最初由 Montreal
大学的机器学习研究者们所开发,用来进行机器学习的计算。
按照官网上的说明,它拥有以下几个方面的特点:
- 与 Numpy, Scipy 的紧密结合
- GPU 加速
- 高效的符号计算
- 速度和稳定性
- 动态生成 C 代码
windows
下,使用 anaconda
安装 theano
的命令为:
conda install mingw libpython
pip install theano
linux
下,使用 anaconda
安装的命令为
conda install theano
安装好之后,还需要安装 Cuda
并进行 GPU
环境的配置,否则是不能利用 GPU
进行计算的,推荐使用 linux/mac
进行配置,具体方法可以参考官网上的配置说明。
查看安装的版本:
In [1]:
import theano
theano.__version__
Out[1]:
'0.7.0.dev-54186290a97186b9c6b76317e007844529a352f4'
查看当前使用的 device:
In [2]:
theano.config.device
Out[2]:
'cpu'
运行测试:
In [3]:
theano.test()
/usr/local/lib/python2.7/dist-packages/theano/misc/pycuda_init.py:34: UserWarning: PyCUDA import failed in theano.misc.pycuda_init
warnings.warn("PyCUDA import failed in theano.misc.pycuda_init")
....................S...............
Theano version 0.7.0.dev-54186290a97186b9c6b76317e007844529a352f4
theano is installed in /usr/local/lib/python2.7/dist-packages/theano
NumPy version 1.10.1
NumPy relaxed strides checking option: True
NumPy is installed in /usr/lib/python2.7/dist-packages/numpy
Python version 2.7.6 (default, Jun 22 2015, 17:58:13) [GCC 4.8.2]
nose version 1.3.7
----------------------------------------------------------------------
Ran 37 tests in 37.919s
OK (SKIP=1)
Out[3]:
<nose.result.TextTestResult run=37 errors=0 failures=0>
这里我已经在本地 Windows
配好了 GPU
的设置,如果没有配好,显示的结果可能不一样。
Windows
下第一次运行可能会显示 DEBUG: nvcc STDOUT
等内容,Just ignore it!