本人用了差不多一周的时间终于在ubuntu14.04的环境下成功安装pycuda-2014.01.
在进行deep learning 过程中,为了提高程序的运行效率,我们通常会借助gpu进行并行计算,本人以前进行过cuda并行计算的研究和学习,所以想借助于cuda对deep learning程序进行加速,但是之前是基于cuda c进行编程,而python最大的特点就是支持的库很多,如numpy,使用起来比c语言更具优势,但在速度方面python不如c,由于pycuda的出现能很好的解决相应的问题。
安装步骤如下:
1.首先安装ubuntu-64bit
2.python-six更新
ubuntu14.04好像自带的python-six版本是1.5.2,但是pycuda-2014.01必须要用到python-six的高级版本,笔者在进行python-six的更新时浪费了很多时间,并且在pycuda的wiki邮件列表里上能找到python-six的问题,但是邮件列表内并没有解决该问题。通过语句
sudo apt-get python-six upgrade
虽然可以对python-six进行更新,但是更新之后系统仍然使用的是python-six1.5版本。可以通过
ThinkCentre-M57e:~/pycuda-2014.1$ python
Python 2.7.6 (default, Mar 22 2014, 22:59