windows下cuda的配置与安装及pycuda的安装

3 篇文章 0 订阅
1 篇文章 0 订阅

windows下cuda的配置与安装

在上一篇博客中安装theano由于要使用cuda加速,所以已经安装过,在这里只是简单介绍一下
nvidia应该算是在2016年炙手可热的公司,股价涨到了100美元以上,主要是依靠gpu在机器学习上的强势计算能力,这主要得益于cuda以及cudnn的加速能力。所以倘若你要进行机器学习工作,gpu加速是不可避免的,theano和caffe以及tensorflow都需要cuda以及cudnn的加速,所以可见cuda安装的必要性
以tensorflow的安装为例,建议参考以下视频
http://www.bilibili.com/video/av7676039/
cuda的安装要注意的就是cuda 7.0之后只有windows 64位系统,并且cudnn并没有windows8及8.1版本,所以要么重装64位系统,要么装cuda6.5之前的版本。cuda解压缩即可,环境变量一般在安装的过程中选择自动添加。cudnn也只需要解压缩即可,把压缩后的三个文件夹(lib include bin)添加到PATH环境变量即可。
cuda的下载地址:https://developer.nvidia.com/cuda-downloads
cudnn的下载地址:https://developer.nvidia.com/cudnn

pycuda的安装

cuda一般是指采用C语言编程实现的GPU cuda架构,pycuda就是指使用python语言实现的cuda架构,同属于nvidia显卡加速并行计算架构。pycuda的安装也比较简单,方法如下:
1.下载whl文件:http://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda
需要选择自己合适的版本,包括python版本,cuda版本,windows 32/64.
2.在cmd下运行pip install pycuda-2014.1+cuda6514-cp27-none-win32.whl,需要注意的是需要把whl文件放到你的home文件下,或者cd在它下载的文件所在的位置即可。
3.测试,测试代码如下:

# (from PyCUDA's documentation)
import pycuda.autoinit
import pycuda.driver as drv
import numpy

from pycuda.compiler import SourceModule
mod = SourceModule("""
__global__ void multiply_them(float *dest, float *a, float *b)
{
  const int i = threadIdx.x;
  dest[i] = a[i] * b[i];
}
""")

multiply_them = mod.get_function("multiply_them")

a = numpy.random.randn(400).astype(numpy.float32)
b = numpy.random.randn(400).astype(numpy.float32)

dest = numpy.zeros_like(a)
multiply_them(
        drv.Out(dest), drv.In(a), drv.In(b),
        block=(400,1,1), grid=(1,1))

assert numpy.allclose(dest, a*b)
print(dest)

输出结果为一个列表:
这里写图片描述
至此,可以开始愉快的学习theano了,theano还是比较适合新手入门的机器学习库,但tensorflow毕竟的google的亲儿子,如果以后准备从事相关工作的话,tensorflow可能更加适合。theano官方的学习文档地址为:
http://deeplearning.net/software/theano/tutorial/index.html#basics

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值