theano gpu配置

来源:互联网 发布:c语言字符串大小写转换 编辑:程序博客网 时间:2024/06/04 18:55

安装theano

Anaconda安装theano可用conda直接安装
  conda install theano


配置.theanorc

生成文件 sudo gedit ~/.theanorc(注意不要漏掉theano前面的一个点)并复制如下内容,然后保存,其中cuda一项的内容为cuda所安装的位置。
  [global]
  floatX=float32
  device=gpu
  [cuda]
  root=/usr/lib/nvidia-cuda-toolkit
  [nvcc]
  flags=-D_FORCE_INLINES

至此theano的安装及配置已经完成,在python里输入import theano,显示使用的GPU设备表示GPU配置成功。

原创粉丝点击