一、概述
由于win10系统针对开源模型的推理训练存在兼容性不佳的情况(通常为Linux运行环境),若要在window系统上跑带GPU环境的Linux系统。这时就需要安装WSL2,当然WSL2的GPU要求也不是完全跟windows10上的一样,根据微软官方文档的要求,在x64机器上安装WSL2需要Windows版本 1903 或以上,内部版本 18362 或以上。
二、安装WSL2
根据微软的官网:安装 WSL | Microsoft Learn
三、WSL2安装CUDA驱动
1、根据CUDA的官网说明:1. NVIDIA GPU Accelerated Computing on WSL 2 — CUDA on WSL 12.3 documentation
2、安装cudnn:developer.nvidia.com/rdp/cudnn-a…
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-11.5(自己检查具体的版本修改路径)/lib64/
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.5(自己检查具体的版本修改路径)/include/
#为更改读取权限:
sudo chmod a+r /usr/local/cuda-11.5(自己检查具体的版本修改路径)/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.5(自己检查具体的版本修改路径)/lib64/libcudnn*
四、验证
1、上述步骤执行完之后,在WSL2的窗口中执行nvidia-smi命令,查看相关配置。若提示nvidia-smi命令无效,代表CUDA未安装成功。
2、nvcc -V验证是否成功
五、注意事项
或许有人遇到在wsl2的系统中运行Docker相关命令时出现以下报错:
Cannot connect to the Docker daemon at unix:///var/run/docker.sock. Is the docker daemon running?
解决办法如下: 1、打开Docker Desktop应用程序,找到设置里的Resource菜单中wsl integration项,重新点击打开相关按钮,重启docker engine即可生效。