在 WSL2 上安装 Tensorflow-GPU

Question

有没有人在 WSL2 上使用 NVIDIA GPU 成功安装 Tensorflow-GPU？我在 WSL2 上安装了 Ubuntu 18.04，但正在努力安装 NVIDIA 驱动程序。任何帮助将不胜感激，因为我迷路了。

Answer 1

所以我刚刚开始运行。

您需要遵循的步骤在这里。总结一下：

1. 注册 windows 内幕程序并获取 windows 的开发版本，以便您拥有最新版本
2. 安装 wsl 2
3. 从 Windows 商店安装 Ubuntu
4. 在 Windows 上安装 wsl 2 cuda 驱动程序
5. 安装 cuda 工具包
6. 安装cudnn（可以从windows下载linux版本，然后将文件复制到linux）
7. 如果您遇到诸如“无法分配内存”之类的内存错误，那么您可能需要增加 wsl 可以获得的内存量
8. 然后安装tensorflow-gpu
9. 祈祷它有效

我一路上遇到的错误：

• 如果第一次打开 ubuntu 时出现错误，则需要在 bios 中启用虚拟化
• 如果您无法运行安装说明中的 ./Blackscholes 示例，则您可能没有正确构建的 Windows！你必须有正确的版本
• 如果在运行 tf 时遇到“无法分配内存”错误，则需要为 wsl 提供更多内存。默认情况下它只访问你的 ram 的一半
  1. 在 Windows 中的用户目录下创建一个 .wslconfig 文件，其中包含所需的内存量。我的看起来像：

[wsl2]
memory=16GB 

运行一些代码后编辑

当我直接在 Windows 上运行时，这要慢得多。我从每 epoch 1 分钟到 5 分钟。我只是要双启动。

Answer 2

我可以确认，由于以下文章，我能够在 WSL2 上不需要 Docker 的情况下完成此工作：

https://qiita.com/Navier/items/cf551908bae707db4258

请务必更新到驱动程序版本460.15，而不是455.41CUDA 文档中列出的版本。

请注意，这不适用于 TCC 模式下的卡（仅限 WDDM）。另外，请确保将文件放置在 Linux 文件系统上（即不要放置在安装驱动器上，如/mnt/c/）。Linux 文件系统上的性能明显更快（这与 WSL 1 与 WSL 2 的实现差异有关；请参阅1、2和3）。

注意：另请参见Keras/Tensorflow 中的类生成器（继承序列）线程安全吗？