当前位置: 首页 > 工具软件 > wsl-guide > 使用案例 >

WSL2 安装、配置 Cuda、pytorch 记录

居飞扬
2023-12-01

WSL2 安装、配置 Cuda、pytorch 记录

本文地址:blog.lucien.ink/archives/532

最近整了张矿卡,为了这盘醋,包了盘饺子。虽然我已经预料到买前深度学习,买后电子竞技,但还是象征性的先配下环境。

安装过程

我使用的是 Windows 11,所以相比 Windows 10,有些命令可能不太适用。

安装显卡驱动

直接去 NVIDIA 官网 下载 Game Ready 驱动即可。

安装 WSL2

个人比较喜欢 Debian,比较轻量级。

wsl --install -d Debian

安装完成之后执行 wsl 进入 Linux,在 Linux 下执行一下 nvidia-smi,输出显卡信息即没有问题。

安装 Miniconda

个人不是很推荐 Anaconda,不赘述。

前往 Miniconda 官网 下载 Miniconda3 Linux 64-bit 版本安装即可。

安装 pytorch-gpu

根据 PyTorch 官网 的提示,执行命令即可。

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

QA

Q: 为什么选择 Windows
A: 目前 Windows 是地球上最好的 Linux 发行版之一,其次它也是能同时满足学习、娱乐的唯一选择。

参考文档

  1. Install Linux on Windows with WSL
  2. CUDA on WSL User Guide
 类似资料: