关于为theano设置GPU

时间:2017-03-15 17:01:28

标签: theano theano-cuda

我在home / .theanorc下配置GPU如下

<%= select_tag(:smoker, options_for_select([['Yes', 1], ['No', 0]])) %>

运行程序会发出以下警告信息,如何解决这个问题?

bash-4.1$ cat .theanorc
[global]
floatX = float32
device = gpu0

[lib]
cnmem = 1

1 个答案:

答案 0 :(得分:0)

可以通过提供链接(https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29)中的步骤来缓解有关cuda后端的警告 这只是一个警告,如果你没有解决它,代码应该运行事件。

对于错误,首先需要确保安装了nvcc编译器并正确设置了路径。假设安装在目录

<强> /usr/local/cuda-7.0

您需要执行以下操作

  

export PATH = / usr / local / cuda-7.0 / bin:$ PATH

  

export LD_LIBRARY_PATH = / usr / local / cuda-7.0 / targets / x86_64-linux / lib:$ LD_LIBRARY_PATH