【AI实战】CUDA 编程入门及开源项目代码分享


在这里插入图片描述

CUDA 简介

CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。

CUDA™是一种由NVIDIA推出的通用并行计算架构, 该架构使GPU能够解决复杂的计算问题。 它包含了 CUDA指令集架构(ISA)以及GPU内部的并行计算 引擎。

开发人员可以使用C语言来为CUDA™架构编 写程序,所编写出的程序可以在支持CUDA™的处理 器上以超高性能运行。(来源:百度百科)

随着显卡的发展,GPU越来越强大,而且GPU为显示图像做了优化。在计算上已经超越了通用的CPU。

如此强大的芯片如果只是作为显卡就太浪费了,因此NVIDIA推出CUDA,让显卡可以用于图像计算以外的目的。

CUDA(Compute Unified Device Architecture)是一个新的基础架构,这个架构可以使用GPU来解决商业、工业以及科学方面的复杂计算问题。

CUDA的SDK中的编译器和开发平台支持Windows、Linux系统,可以与Visual Studio2005,2008,2010集成在一起。

官方文档

简单的代码

将大小为N的两个向量 A 和 B 相加,并将结果存储到向量 C 中:

// Kernel definition
__global__ void VecAdd(float* A, float* B, float* C)
{
    
    
    int i = threadIdx.x;
    C[i] = A[i] + B[i];
}

int main()
{
    
    
    ...
    // Kernel invocation with N threads
    VecAdd<<<1, N>>>(A, B, C);
    ...
}

以下代码将大小为 N x N 的两个矩阵 A 和 B 相加,并将结果存储到矩阵 C 中:

// Kernel definition
__global__ void MatAdd(float A[N][N], float B[N][N],
                       float C[N][N])
{
    
    
    int i = threadIdx.x;
    int j = threadIdx.y;
    C[i][j] = A[i][j] + B[i][j];
}

int main()
{
    
    
    ...
    // Kernel invocation with one block of N * N * 1 threads
    int numBlocks = 1;
    dim3 threadsPerBlock(N, N);
    MatAdd<<<numBlocks, threadsPerBlock>>>(A, B, C);
    ...
}

开源学习的代码

这里分享一些开源项目,以便更好的学习。

根据各自的需求阅读对应的源代码。

能自己动手写就动手写,多写才是王道。

参考

猜你喜欢

转载自blog.csdn.net/zengNLP/article/details/127722076