使用theano时报错:ImportError : No module named pygpu

描述

在使用theano调用GPU时,老版本的调用方法会用到pygpu这个包,但是现在theano并不会内置这个包,所以就会出现以下错误:

ImportError : No module named pygpu

解决方法

  1. 下载并编译安装libgpuarray
# 下载源码
git clone https://github.com/Theano/libgpuarray.git
# 编译安装
cd libgpuarray
mkdir Build && cd Build
cmake .. -DCMAKE_BUILD_TYPE=Release
make -j 6 && sudo make install
cd ..
  1. 安装 pygpu
# This must be done after libgpuarray is installed as per instructions above.
python setup.py build
python setup.py install
  1. 执行链接到系统环境中
# python setup.py build_ext -L MYPREFIX/lib−IMY_PREFIX/include
# 注意这里的My_PREFIX是你安装的libgpuarray库目录。默认是在/usr/local下
python setup.py build_ext -L /usr/local/lib−IMY_PREFIX/include

# 更新链接
sudo ldconfig

# 编辑环境变量文件
vim ~/.bashrc
# 添加内容如下:
export LIBGPUARRAY_PATH=/usr/local
export PATH=LIBGPUARRAY_PATH/lib:$PATH
# 立即生效
source ~/.bashrc
  1. 测试
DEVICE="cuda" python -c "import pygpu;pygpu.test()"
手机上阅读

本文由 kivii创作, 采用 知识共享署名4.0 国际许可协议进行许可
本站文章除注明转载/出处外,均为本站原创或翻译,转载前请务必署名
原文地址:《使用theano时报错:ImportError : No module named pygpu》

 最后一次更新于2019-09-14

0 条评论

添加新评论

Markdown is supported.