python - 在 virtualenv 中使用 CUDA 为 Theano 服务

标签 python cuda virtualenv theano lasagne

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

运行一些代码,我得到一个 ImportError: dnn not available,据我所知,这是由于缺少 CUDA header 而导致的,尽管它安装在 /usr/local/cuda- 6.5/targets/x86_64-linux/include/cudnn.h.

theano documentation 之后我设置了环境变量 CUDA_ROOT=/usr/local/cuda-6.5,这会产生一个额外的错误:

错误(theano.sandbox.cuda):编译cuda_ndarray.cu失败:libcublas.so.6.5:无法打开共享对象文件:没有这样的文件或目录

有什么方法可以将这个 CUDA 库包含到 virtualenv 中吗?

最佳答案

终于找到答案了。除了环境变量之外,我还忘记在我的主目录中配置 .theanorc,例如 theano documentation。建议:

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

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

还是谢谢!学习了仔细阅读文档的类(class)......

关于python - 在 virtualenv 中使用 CUDA 为 Theano 服务,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33083852/

相关文章:

python:使用元组的多个变量

python - 从YouTube channel 逐页获取所有视频

linux - 是否可以在应用程序之间共享 Cuda 上下文?

python - 安装的 Virtualenv 和激活 virtualenv 不起作用

python - Python 3.5 (virtualenv) 中没有持久的命令历史记录

python - 打印目标函数 Pyomo 项的求解器结果

python - 执行 python-scrappy 模块时出错

CURAND - 设备 API 和种子

cuda - CUDA内存是如何管理的?

python - 如何在 Ubuntu bash 中使用 Python 虚拟环境?