在gtx1080上运行theano环境搭建



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/

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值