WSL安装CUDA与CUDA踩坑

先说CUDA的坑

NVCC的版本需要小于Nvidia-smi的版本，不然用nvcc编译的程序无法正确运行，关键是为什么都没有报错啊啊啊啊啊啊！我被这鬼东西坑了两个小时啊，找不到问题所在啊！！！

在安装NVCC之前一定要先用nvidia-smi，查看CUDA版本，不要安装比nvidia-smi显示的更高的CUDA版本。

WSL安装CUDA

参考网站

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-4

这里的版本一定要注意，不要乱安装，官方这里给的是最新版12.4，要根据自己的机器修改这个值。

然后设置环境变量

sudo vim ~/.bashrc

最后追加(这里的版本也要根据自己安装的来)

export PATH=/usr/local/cuda-12.4/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.4/targets/x86_64-linux/lib/:$LD_LIBRARY_PATH

使配置生效

source ~/.bashrc

查看是否成功

nvcc -V

卸载CUDA

sudo apt-get --purge -y remove 'cuda*'
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*

reference:

1. https://docs.nvidia.com/cuda/wsl-user-guide/index.html#getting-started-with-cuda-on-wsl-2