python - 在 virtualenv 中使用 CUDA 服务 Theano

Question

由于缺乏 root 权限，我在 Ubuntu 14.04 的 python3 virtualenv 中安装了 Theano 和 Lasagne。

ImportError: dnn not available运行一些代码，我得到一个/usr/local/cuda-6.5/targets/x86_64-linux/include/cudnn.h.

根据theano 文档，我设置了 environment variable CUDA_ROOT=/usr/local/cuda-6.5，这会产生一个额外的错误：

ERROR (theano.sandbox.cuda): Failed to compile cuda_ndarray.cu: libcublas.so.6.5: cannot open shared object file: No such file or directory

有没有办法将此 CUDA 库包含到 virtualenv 中？

Answer 1

终于找到了答案。除了环境变量之外，我忘记.theanorc在我的主目录中配置，就像theano 文档建议的那样：

$ cat /home/user/.theanorc
[global]
floatX = float32
device = gpu0

[cuda]
root = /usr/local/cuda-6.5

不管怎么说，还是要谢谢你！吸取教训仔细阅读文档...