WSL2 CUDA踩坑指南

必要环境

1. 支持CUDA的显卡。。。（我的是1660Ti）

2.WSL2 （没有升级的参考https://docs.microsoft.com/en-us/windows/wsl/install-win10）

3. 官方推荐win10  Build 20150 or higher（ 但是我的是21H1 19043，好像也可以），实测是这样的。。在预览计划中选择dev通道，更新最新版本的windows

安装步骤：

Nvidia预览版驱动

去这个网站下：https://developer.nvidia.com/cuda/wsl

 

下面的操作都在WSL2中了。我的是Ubuntu2004

添加源

apt-key adv --fetch-keys http://developer.download.nvidia.cn/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo sh -c 'echo "deb http://developer.download.nvidia.cn/compute/cuda/repos/ubuntu2004/x86_64 /" > /etc/apt/sour
ces.list.d/cuda.list'
sudo apt-get update

原nVidia指南网站上有个坑，除了修改对应系统版本外，还要修改.com -> .cn，不然update会失败，参考：https://github.com/microsoft/WSL/issues/5682#issuecomment-788744572

安装cudatools

sudo apt-get install -y cuda-toolkit-11-0

因为我的win10版本比官方推荐低，本来以为会失败，没想到到这里一切顺利。。。

这一步有点漫长，可以去休息一会儿了~

运行示例程序

首先切换到目录

cd /usr/local/cuda/samples/4_Finance/BlackScholes

然后执行sudo make，构建程序，输出如下：

貌似有点问题。。

执行程序

./BlackScholes

果然有问题：

改日再战。。

更新系统版本后就没有问题了：

实测pytorch也可以正常启用cuda：

参考网站：

1. 微软： https://docs.microsoft.com/en-us/windows/win32/direct3d12/gpu-cuda-in-wsl

2. Nvidia: https://docs.nvidia.com/cuda/wsl-user-guide/index.html