python中chainer的gpu环境配置问题:
前gpu计算的显卡为最新的GeForce 系列 的GTX1080,这款显卡对应的驱动为367.27和367.18,而这两个驱动对应的sm为61,
其次说cuda的安装,chainer支持的cuda的版本为6.5-7.5版本,而最新的cuda7.5对应的sm为5.3,与显卡的sm不兼容。
最新版的cuda8.0的sm可能为61,但是这样的话chainer就不支持了
导致现在是以目前显卡的情况不能用cuda的chainer
caffe是否可用呢,这个也没有试,不知道会不会有相同的问题,试着装caffe,但是没装成功,这个目前也不想折腾了。
下面的思路是考察theano对最新显卡gtx1080 的支持情况,theano也是运行在cuda之上的,目前的情况是cuda只能用最新的8.0,需要查一下theano对cuda8.0的支持情况。
利用theano的rnn自动生成文本,github地址:
https://github.com/dennybritz/rnn-tutorial-gru-lstm#jupyter-notebook-setup
内含有理论说明的链接
在gtx1080上运行theano环境搭建:
先装驱动
装cuda8
在装驱动
装python的依赖包
装theano
详细参考:
http://www.chioka.in/how-to-setup-theano-to-run-on-gpu-on-ubuntu-14-04-with-nvidia-geforce-gtx-780/
https://devtalk.nvidia.com/default/topic/938988/cuda-setup-and-installation/-solved-cuda-8-0-on-ubuntu-16-04-gpu-not-available/