CUDA编程练习（一）环境创建

原创

已于 2024-07-26 15:35:38 修改

· 406 阅读

0 ·

版权

文章标签：

#python #c++ #jupyter

于 2024-05-28 22:06:39 首次发布

1.1 安装nvcc4jupyter
建议找GitHub官方教程，更新太快了

!pip install nvcc4jupyter
%load_ext nvcc4jupyter

1.2 示例程序

%%cuda
#include <iostream>
int main(){
    std::cout << "Hello cuda!\n";
    return 0;
}

1.3 显卡使用情况

!nvidia-smi

1.4 显卡的各个参数

%%cuda
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
int main()
{
    int deviceCount;
    cudaGetDeviceCount(&deviceCount);
    int dev;
    for (dev = 0; dev < deviceCount; dev++)  // 就一张显卡
    {
        cudaDeviceProp deviceProp;
        cudaGetDeviceProperties(&deviceProp, dev);
        if (dev == 0)
        {
            if (/*deviceProp.major==9999 && */deviceProp.minor = 9999&&deviceProp.major==9999)
                printf("\n"