WSL2中使用Windows的CUDA

1.Windows安装cuda驱动（参见https://developer.nvidia.com/cuda/wsl）

2.Ubuntu中安装cuda-tookit

sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub

sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"

sudo apt-get update

toolkit的版本一定要选择cuda版本对应的
sudo apt-get install -y cuda-toolkit-11-6

查看自己显卡驱动的方法可以是在windows11下面的powershell使用nvida-smi就可以看到了。

然后：

vim ~/.bashrc
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PAT

source ~/.bashrc

如此即可看到：