WSL2+TF

文章目录

• 前言
• 一、安装wsl
• 二、安装显卡驱动
• 三、安装docker和nvidia container toolkit
• 四、安装cuda（可选）
• 五、下载及运行tf
• 总结

---

前言

本文记录如何在wsl2上安装tf。

windows: 10

wsl: ubuntu 18.04

tf: tensorflow:latest-gpu（docker映像）

参考文档：

• CUDA on WSL :: CUDA Toolkit Documentation (nvidia.com)
• Docker  |  TensorFlow (google.cn)

wsl2+tf docker存在缺点，自行判断是否适合使用：

1. wsl特性和实机linux不同，安装步骤不同，出bug不好调；
2. tf docker映像没有安装太多python库比如tqdm，若有依赖需要自己维护容器；
3. tf docker需要以root运行，以获得安装依赖的权限。

也有优点：

1. 不用装cuda——来自tf docker的支持。
2. 不用双系统。

---

一、安装wsl

因时间跨度原因，不记录细节。