配置过程
0前言
这一篇文章之前配置过windows不用wsl来配置cuda编译环境,地址:点击这里,不用wsl的缺点显然,不可以使用cuda gdb来调试代码,每次都是编译成可执行文件一遍遍测试代码十分累,于是这篇文章诞生。
配置这些是因为本人一直在手写一些算子,希望可以与cublas等等库达到差不多的性能来学习到这其中的精髓,我的项目地址是:点击这里,如果觉得有帮助,记得点哥star。
1 下载wsl2
这里可以参考我给的,当然也可以去看csdn上更加详细的图形化的安装步骤。
step1:启用 WSL 和虚拟化功能
按 Win + X,选择 Windows PowerShell (管理员),输入以下命令来启用 WSL 和虚拟机平台功能:
dism.exe /online /enable-feature /featurename:Microsoft-Windows-Subsystem-Linux /all /norestart
dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart
step2:启用 Hyper-V
WSL2 需要 Hyper-V 来支持虚拟化, 输入以下命令:
dism.exe /online /enable-feature /featurename:Microsoft-Hyper-V-All /all /norestart
step3:重启电脑
step4: Microsoft Store安装你要的ubuntu版本,这里我下载的是22.04。 (tips:这里我用了玩游戏用的加速器,里面有微软商店的加速下载服务)。
step5:测试成功没有, 在win + R打开的cmd输入以下指令,看到进入到ubuntu的terminal就是成功了。
wsl
2 下载vscode(这一步在windows中下载,不在wsl2中下载)
step1:完成vscode下载安装后,依次点击左下角的“设置”,“配置文件”,“创建配置文件”,然后点击“创建“。(这里是为了让cuda的编程环境能够与其他项目隔离)
step2:此时点击图中左下角的大于小于><符号,如图
step3:在此处的终端发现可以输入linux指令。完成!
3 安装CUDA 和 TensorRT
这里的每一步都与正常的ubuntu下安装cuda和tensoRT一致,所以下面你在安装的过程遇到问题都可以在csdn上找到其他博主的解决方案。截至25年1月3日,我发现官方已经出了,Getting Started with CUDA on WSL 2
step1 :先终端输入
nvidia-smi
在右上角可以看到自己当前电脑支持的最高cuda版本。
step2:下载前先 在 Ubuntu 中更新软件包列表
sudo apt update
sudo apt upgrade -y
step3:从NVIDIA官网下载并安装CUDA:CUDA Toolkit。
[注意]
CUDA 工具包的安装说明可以在每个安装程序的 CUDA 工具包下载页面中找到。但请勿在 WSL 2 下选择“cuda”、“cuda-12-x”或“cuda-drivers”。
step4:下载TensorRT,TensorRT下载界面,这里一定要看官方的,我按照官方的下载没有任何问题。
step5: 步骤3和4完成后,再执行一次step2更新软件包列表,这里把安装好的cuda和tensorRT都加入到环境变量bashrc中(参考其他博主怎么把cuda和tensorRT加入到环境变量就行,相关文章很多),可以看到如下说明成功。
step6:用以下两个指令看是否都下载完成
4 配置环境变量
输入
vim ~/.bashrc
参考我的,具体的路径你需要自己根据进行修改,完成后按ESC ,然后输入
:wq
退出vim,然后输入
sourch ~/.bashrc
保存环境变量的修改
5 创建cuda + tensorRT项目
参考的github,有很多相关项目。点击这里,如果项目对你有帮助,记得star