在wsl2中安装CUDA

1、先根据我之前的教程把wsl1升级到wsl2

wsl1升级到wsl2_夕阳之后的黑夜的博客-CSDN博客

2、打开Ubuntu,输入

 uname -r

确定内核,安装CUDA需要内核 4.19.121 或更高。

3、在Windows 预览体验计划注册账号Windows Insider

4、设置->Windows 预览体验计划,登录刚刚注册的账号,并加入Dev通道,点击确定。

5、更新window11

因为安装wsl2和cuda对操作系统版本有要求,所以干脆直接升级到win11

在wsl2中安装CUDA_第1张图片

6、下载驱动,因为我的GPU是Quadro,所以选择Quadro

CUDA on WSL | NVIDIA Developer

下载完后直接在window上双击安装,安装完后又会跳出一个安装界面

在wsl2中安装CUDA_第2张图片

直接精简安装就行。

 在wsl2中安装CUDA_第3张图片

安装完成。

7、现在就可以打开Ubuntu开始安装cuda了,依次输入下列命令

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install -y cuda-toolkit-11-4

8、测试

$ cd /usr/local/cuda-11.4/samples/4_Finance/BlackScholes
$ su root
$ make BlackScholes
$ ./BlackScholes

测试成功

在wsl2中安装CUDA_第4张图片

9、添加环境变量

vim ~/.bashrc
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc

10、输入

 nvcc -V

结果

在wsl2中安装CUDA_第5张图片

你可能感兴趣的:(WSL,ubuntu,python,1024程序员节)