【发布时间】:2016-06-07 19:03:34
【问题描述】:
我正在尝试在 aws 实例中导入 theano 库以使用 GPU。我已经使用 boto 编写了一个 python 脚本来自动化 aws 设置,它本质上将从我的本地机器对实例执行 ssh,然后在我执行“python -c 'import theano'”的地方启动一个 bash 脚本启动 GPU。但我收到以下错误:
错误(theano.sandbox.cuda):编译 cuda_ndarray.cu 失败:libcublas.so.7.5:无法打开共享对象文件:没有这样的文件或 目录
当我尝试直接在实例命令 shell 中导入 theano 模块时,它会自动开始使用 GPU。
使用 gpu 设备 0:GRID K520(CNMeM 已禁用)
我想我错过了通过我的自动化 python 脚本导入时必须进行的其他一些导入。可能的解决方案是什么?
【问题讨论】:
-
可能是环境问题。尝试 python "import os; print(os.eviron["PATH"])" 看看 "/usr/local/cuda/bin" 是否在 PATH 中。
-
还要检查
LD_LIBRARY_PATH以确保可以在运行时找到 CUDA 库。 -
就我而言,这是由我的 CUDA 安装策略引起的问题。当我尝试安装 libcublas.so.10 时,它正在创建 libcublas.so.11。在我的安装脚本中使用
DEBIAN_FRONTEND=noninteractive apt -y install cuda-10-1解决了这个问题。
标签: python amazon-web-services cuda gpu theano