有没有办法在 windows 电脑上 运行 RAPIDS?

Is there a way to run RAPIDS on windows pc?

我正尝试在 windows 计算机上 运行 Nvidia 急流,但没有成功。我已经为 windows 安装了 docker 桌面并下载了 rapids 图像。已安装 Cuda 10.0,但未安装 Nvidia-container-toolkit。我没能做到 运行。有什么想法或指导吗?

20 年 9 月 6 日更新: Wesley 撰写的答案在最新的 Windows WSL2 Insider Preview 中是准确的。我没有修改这个答案,而是对他的答案进行了编辑。


没有。现在,RAPIDS 需要一个 Linux 主机。这是在 NVIDIA 最近的一次研讨会上提出的。还提到 RAPIDS 不能与 WSL 一起使用。它可能 与 WSL 版本 2 一起工作,但我没有尝试过,也不知道有人这样做。

唯一的选择是,如果您可以将 GPU 分配给 Windows 主机上的 Linux 虚拟机。这 可能 但足够复杂,双引导是更好的解决方案。

我不确定是否有人对原始问题给出了更明确的 'updated' 答案。此时(2020 年 8 月),答案是“是!”。在 Windows 10 上,您绝对可以 运行 WSL2 中的 RAPIDS,但要满足一些条件:

要求

  • 您必须在 Windows 子系统中为 Linux 版本 2 (WSL2) 使用 RAPIDS;

  • Windows10个版本 2004(OS 内部版本 202001.1000 或更高版本) 您必须注册才能获得 Windows Insider Preview 版本,特别是 Developer Channel。这是 WSL2 VM 具有 GPU 访问权限所必需的。 https://insider.windows.com/en-us/

  • CUDA SDK v11.1 中的 CUDA 版本 455.41 您必须使用特殊版本的 NVIDA CUDA 驱动程序(我正在使用) 您必须从 NVIDIA 的站点通过特殊下载获得。你必须 加入 NVIDIA Developer Program 以获取版本 -- 然后搜索 'WSL2 CUDA Driver',它应该会引导您找到它。

设置

  1. 安装 windows 的开发者预览版。请务必单击 'update' 中的复选框,该复选框也会安装其他推荐的更新。
  2. 从 NVIDIA 开发者计划安装 windows CUDA 驱动程序
  3. 通过启用“虚拟机平台”可选功能来启用 WSL 2。您可以在此处找到更多步骤 https://docs.microsoft.com/en-us/windows/wsl/install-win10
  4. 从 Windows 商店安装 WSL(Ubuntu-20.04 确认工作)
  5. 在 WSL VM 上安装 python,使用 Anaconda
  6. 进行测试
  7. 安装Rapids AI(最好在你有数百个其他包之前安装这个 'conda' 试图 self-consistently 与急流依赖图协调——你可以以后总是通过 pip 或 conda 安装额外的 python 包。)

执行此操作后,如果您启动 ipython...

Python 3.8.3 (default, May 19 2020, 18:47:26)
Type 'copyright', 'credits' or 'license' for more information
IPython 7.17.0 -- An enhanced Interactive Python. Type '?' for help.

>>> import cuml
>>> cuml.__version__
'0.15.0'

>>> import cudf
>>> cudf.__version__
'0.15.0'

>>> import dask_cudf
>>> dask_cudf.__version__
'0.15.0'

>>> import cupy
>>> cupy.__version__
'7.8.0'

...您可以使用 RAPIDS AI。