让Theano使用GPU

Question

设置Theano与我的显卡一起使用时遇到了一些麻烦 - 我希望你们能帮我一臂之力.

我之前使用过CUDA,并且正确安装,因为运行Nvidia Nsight是必要的.但是,我现在想将它与PyDev一起使用,并且在http://deeplearning.net/software/theano/install.html#gpu-linux上的"使用GPU"部分之后遇到了一些问题.

第一个是非常基础的,那就是如何设置环境变量.它说我应该' 定义一个$ CUDA_ROOT环境变量 '.有几位消息人士表示要在我的主目录中创建一个新的".pam_environment"文件.我做了这个并写了以下内容:

CUDA_ROOT = /usr/local/cuda-5.5/bin
LD_LIBRARY_PATH = /usr/local/cuda-5.5/lib64/lib

我不确定这是否正是它必须写的方式 - 如果这是一个基本问题,请道歉.如果我能确认这确实是写它的正确位置,那将会有所帮助.

第二个问题在本教程的以下部分.它说" 更改设备选项以命名计算机中的GPU设备 ".显然这与THEANO_FLAGS和.theanorc有关,但我无法找到它们是什么:它们是文件吗？如果是这样我在哪里可以找到它们？该教程似乎假设了一些我没有的知识!

感谢您花时间阅读本文:非常感谢任何和所有答案 - 我现在非常完全陷入困境!

Answer 1

在Linux/OSX上:

编辑或创建文件~/.theanorc.该文件应包含:

[global]
floatX = float32
device = gpu0

[nvcc]
fastmath = True

[cuda]
root=/usr/local/cuda-5.5/  
# On a mac, this will probably be /Developer/NVIDIA/CUDA-5.5/

您需要将cuda添加到$ LD_LIBRARY_PATH变量中.如果您正在运行eclipse,您可以转到项目属性>解释器>配置和解释器...>环境,然后添加一个指向您的cuda lib文件夹的LD_LIBRARY_PATH变量(可能是/Developer/NVIDIA/CUDA-5.5/lib64 )

现在当你导入theano时,它应该打印一条关于找到gpu的消息.您可以在http://deeplearning.net/software/theano/tutorial/using_gpu.html上运行测试代码,看看它是否正在使用gpu.

Answer 2

THEANO_FLAGS是一个环境变量，.theanorc 是一个配置文件。您可以使用这两种机制来配置 Theano。此处对此进行了描述。

我从未听说过 .pam_environment 文件。另外，您不应该只覆盖 的值LD_LIBRARY_PATH，而应像这样附加/前置它：

LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-5.5/lib64/lib

对于 Theano，如果定义了CUDA_ROOT，则不需要修改LD_LIBRARY_PATH，所以我只需删除最后一行。

CUDA_ROOT通常，如果您的 shell 是 bash，人们会在 .bashrc 文件中定义 env 变量，如下所示：

export CUDA_ROOT=/usr/local/cuda-5.5/bin

仅当您注销并再次登录时，才会使用对 .bashrc 的更改。