原因
在windows10/11上docker基于WSL运行,想使用gpu需使用WSL2,并在WSL2里安装驱动,nvcc -V以及nvidia-smi都有输出即可。
cuda本机安装后似乎不需要额外安装了,如果需要直接apt安装,用nvcc -V验证是否可以使用cuda,cuda版本建议去Nvidia的窗口查看支持版本,不要安装错了。安装错误需要先卸载。我的win11一直安装cuda不成功 是因为Nsight VSE(Visual Studio Edition)与VS2019冲突,卸载VS就行了,Visual Studio Intergration也不要安装。
安装WSL2
若想使用WSL2,需要Windows版本号大于2004,用windows易升即可。Windows里需要启动WSL功能。当然Hyper-V或虚拟机选项也要打开。这里不再赘述。这时候重启。
输入命令:
wsl --set-default-version 2
之后就安装成功,在Microsoft Store里搜索自己想要的版本,安装。
之后使用wsl -l -v
验证是否为2版本。
安装NVIDIA 驱动
安装ubuntu-drivers-common后输入ubuntu-drivers devices
,有输出就安装对应版本。
我是没有输出,不代表无法使用。去