VScode配置cuda开发环境

因为课程学习需要,配置英伟达的cuda环境做高性能计算的测试。

1.cuda的安装

首先要去英伟达的官网下载安装cuda,安装路径可以自由选择,但是为了方便操作建议直接全部默认安装。

可以现在命令行执行nvidia-smi.exe查看目前显卡支持的最高cuda版本,右上角的cuda version就是我们要确认的版本,下载的cuda不应该比这个版本新。

cuda下载入口:CUDA Toolkit 12.1 Update 1 Downloads | NVIDIA Developer 

进去后选择对应的平台和版本

2.检查环境配置

正常情况下安装完成后是会自动配置环境变量的,但是为了保险我们还是去命令行看一下

 执行nvcc -V,若可以正常显示版本号等信息,说明正常安装。

3.配置vscode

cuda编程用的是.cu文件,我这里为了操作简单用了coderunner插件,配置.cu文件的编译

点击小齿轮选择扩展设置

 

我们直接在settings.json中编辑

 在"code-runner.executorMap": {}中加入cu文件的编译语句

"cu": "cd $dir; nvcc $fileName -o $fileNameWithoutExt.exe -I'C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.7\\include\\CL' && $dir$fileNameWithoutExt",
//请设置和你的安装路径一致的地址

然后我们在.cu文件的右上角点击运行符号选择用coderunner运行

下面是一个测试小程序

#include <iostream>
#include <math.h>
__global__ 
void add(int n, float *x, float *y)
{
    for (int i = 0; i < n; i++)
    y[i] = x[i] + y[i];
}

int main(void)
{
    int N = 1<<25;
    float *x, *y;

    cudaMallocManaged(&x, N*sizeof(float));
    cudaMallocManaged(&y, N*sizeof(float));

    for (int i = 0; i < N; i++) {
    x[i] = 1.0f;
    y[i] = 2.0f;
}

    add<<<1, 1>>>(N, x, y);

    cudaDeviceSynchronize();

    float maxError = 0.0f;
    for (int i = 0; i < N; i++)
    maxError = fmax(maxError, fabs(y[i]-3.0f));
    std::cout << "Max error: " << maxError << std::endl;

    cudaFree(x);
    cudaFree(y);

    return 0;
}

正常运行结果如下

这里可能会返回一个错误,大意是在path找不到cl.exe

如果是这个情况我们可以添加一个path系统环境变量

C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Tools\MSVC\14.33.31629\bin\Hostx64\x64
//注意这里要找到你电脑上的visual studio安装路径,找到这个文件,bin目录下面可能有两个文件夹,但是cuda运行一定要选择x64

然后重启你的vscode,再用coderunner就能正常运行啦

  • 19
    点赞
  • 38
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 13
    评论
配置 CUDA 环境通常需要以下步骤: 1. 安装 NVIDIA 驱动程序:首先,需要确认您的显卡型号并下载对应的 NVIDIA 驱动程序。然后,通过命令行或图形界面进行驱动程序的安装。 2. 安装 CUDA Toolkit:访问 NVIDIA 官方网站下载 CUDA Toolkit 的安装包。选择与您显卡驱动程序兼容的版本,并按照安装向导进行安装。安装完成后,CUDA 相关程序将会被安装到默认目录下。 3. 设置环境变量:为了在命令行能够使用正确版本的 CUDA,需要设置相关的环境变量。打开终端,编辑或创建一个 `.bashrc` 或 `.bash_profile` 文件,并加入以下内容: ``` export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 然后运行以下命令使环境变量生效: ``` source ~/.bashrc ``` 4. 配置 VSCode 插件:打开 VSCode,按下 `Ctrl + P` 输入 `ext install`,然后输入 `TensorFlow Snippets` 并安装该插件。 5. 配置 `tasks.json` 文件:在 VSCode 的文件菜单选择 `Tasks -> Configure Tasks`,然后选择 "Create tasks.json file from template",并选择 "Others"。在新打开的 `tasks.json` 文件,添加以下代码: ```json { "version": "2.0.0", "tasks": [ { "label": "Compile CUDA", "type": "shell", "command": "nvcc", "args": ["${file}", "-o", "${fileDirname}/${fileBasenameNoExtension}"], "group": { "kind": "build", "isDefault": true } } ] } ``` 6. 配置 `launch.json` 文件:在 VSCode 的文件菜单选择 "Debug -> Add Configuration",然后选择 "C++ (GDB/LLDB)"。修改新打开的 `launch.json` 文件的 `"program"` 字段为 `${fileDirname}/${fileBasenameNoExtension}`。 至此,您已成功配置了 Linux 系统上的 VSCode CUDA 环境。现在,您可以通过将 CUDA 代码保存为 `.cu` 文件后按下 `Ctrl + Shift + B` 来进行 CUDA 编译,或者选择 "Run" 菜单的 "Start Debugging" 来进行 CUDA 调试。
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

哆啦叮当

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值