其实Theano下配置GPU和cudnn很简单,也很方便。
就只要cd到/home当前用户的文件夹,其实也就是终端刚打开时的路径。
gedit ~/.theanorc
添加下面内容就可以:
[global]
floatX=float32
device=gpu
[nvcc]
fastmath=True
So easy!!!
本文介绍了一种简单快捷的方法来配置Theano使用GPU及cudnn加速计算。只需要编辑~/.theanorc文件并添加特定参数即可实现,具体包括设置浮点数类型为float32、指定设备为GPU,并启用nvcc的fastmath选项。
其实Theano下配置GPU和cudnn很简单,也很方便。
就只要cd到/home当前用户的文件夹,其实也就是终端刚打开时的路径。
gedit ~/.theanorc
添加下面内容就可以:
[global]
floatX=float32
device=gpu
[nvcc]
fastmath=True
So easy!!!
3572
9622
332

被折叠的 条评论
为什么被折叠?