Theano下怎么配置GPU和cudnn加速

来源:互联网 发布:ubuntu交叉编译器 编辑:程序博客网 时间:2024/05/21 05:05

其实Theano下配置GPU和cudnn很简单,也很方便。

就只要cd到/home当前用户的文件夹,其实也就是终端刚打开时的路径。

gedit ~/.theanorc

添加下面内容就可以:

[global]floatX=float32device=gpu[nvcc]fastmath=True

So easy!!!

0 0