tensorflow - 在 WSL2 上安装 Tensorflow-GPU

Question

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

Answer 1

所以我刚刚开始运行。
您需要遵循的步骤是 here .总结一下:

注册 windows Insider 计划并获取 windows 的开发版本，以便您拥有最新版本

安装 wsl 2

从 Windows 商店安装 Ubuntu

在 Windows 上安装 wsl 2 cuda 驱动程序

安装 cuda 工具包

安装cudnn(你可以从windows下载linux版本，然后将文件复制到linux)

如果您遇到诸如“无法分配内存”之类的内存错误，那么您可能需要增加 wsl 可以获得的内存量

然后安装 tensorflow-gpu

祈祷它有效

我一路上遇到的错误:

如果第一次打开 ubuntu 时出现错误，则需要在 bios 中启用虚拟化

如果您无法运行安装说明中的 ./Blackscholes 示例，则您可能没有正确构建的 Windows!您必须拥有正确的版本

如果在运行 tf 时遇到“无法分配内存”错误，则需要为 wsl 提供更多内存。默认情况下它只访问你的 ram 的一半

在 Windows 中的用户目录下创建一个 .wslconfig 文件，其中包含所需的内存量。我的看起来像:

[wsl2]
memory=16GB 

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