Installation

默认你已经成功安装 WSL2 和某个 Linux 发行版(如 Ubuntu24.04)。

  1. 安装 Nvidia 显卡 Windows 版驱动,该驱动会自动“映射”进 WSL2。系统会自动在 WSL2 的 /usr/lib/wsl/lib/ 目录下生成 libcuda.so 等文件。 不要在 WSL2 中安装 Linux 版本的显卡驱动!,否则会覆盖映射进来的 Windows 版驱动,导致潜在的问题。

  2. 根据需求安装所需版本的 CUDA Toolkit。 如果你只是希望使用 Pytorch,不需要编译自定义 CUDA 算子,那么无须手动安装 CUDA Toolkit,按照 Pytorch 官方文档安装即可,Pytorch 会自带所需的 CUDA 运行时库。

Test

To be continued…