cuda生成随机数-完美版

最新推荐文章于 2025-07-11 11:07:36 发布

GPU小码农

最新推荐文章于 2025-07-11 11:07:36 发布

阅读量5.3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： CUDA

本文链接：https://blog.youkuaiyun.com/zjc_game_coder/article/details/72553094

CUDA 专栏收录该内容

17 篇文章

订阅专栏

本文提供了一个使用CUDA进行大规模随机数生成的示例代码。该示例通过curand库在GPU上初始化状态并生成随机数，适用于高性能计算场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

能用谷歌就不要用百度，真的，对自己好一点，效率高一点。

下面是一个cuda中如何产生随机数的demo，改改可用。我已经用了:D。

int n = 200;
using namespace std;

__device__ float generate( curandState* globalState, int ind ) 
{
    //int ind = threadIdx.x;
    curandState localState = globalState[ind];
    float RANDOM = curand_uniform( &localState );
    globalState[ind] = localState;
    return RANDOM;
}

__global__ void setup_kernel ( curandState * state, unsigned long seed )
{
    int id = threadIdx.x;
    curand_init ( seed, id, 0, &state[id] );
}

__global__ void kernel(float* N, curandState* globalState, int n)
{
    // generate random numbers
    for(int i=0;i<40000;i++)
    {
        int k = generate(globalState, i) * 100000;
        while(k > n*n-1)
        {
            k-=(n*n-1);
        }
        N[i] = k;
    }
}

int main() 
{
    int N=40000;

    curandState* devStates;
    cudaMalloc ( &devStates, N*sizeof( curandState ) );

    // setup seeds
    setup_kernel <<< 1, N >>> ( devStates,unsigned(time(NULL)) );

    float N2[40000];
    float* N3;
    cudaMalloc((void**) &N3, sizeof(float)*N);

    kernel<<<1,1>>> (N3, devStates, n);

    cudaMemcpy(N2, N3, sizeof(float)*N, cudaMemcpyDeviceToHost);

    for(int i=0;i<N;i++)
    {
        cout<<N2[i]<<endl;
    }

    return 0;
}