有没有一种方法可以在Windows PC上运行RAPIDS?

时间:2019-10-30 10:21:04

标签: gpu rapids

我试图在Windows计算机上运行Nvidia Rapids,但没有任何运气。我已经为Windows安装了docker桌面并下载了Rapids映像。已安装Cuda 10.0,但未安装Nvidia-container-toolkit。我无法使其运行。有什么想法或指导吗?

3 个答案:

答案 0 :(得分:7)

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

要求

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

  • Windows 10版本 2004(操作系统内部版本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驱动程序”,它将引导您进入它。

设置

  1. 安装Windows的开发人员预览版。确保单击“更新”中的复选框,该复选框也将安装其他建议的更新。
  2. 从NVIDIA开发人员计划安装Windows CUDA驱动程序
  3. 通过启用“虚拟机平台”可选功能来启用WSL 2。您可以在这里https://docs.microsoft.com/en-us/windows/wsl/install-win10
  4. 中找到更多步骤
  5. 从Windows应用商店安装WSL(已确认Ubuntu-20.04正常运行)
  6. 在WSL VM上安装经过Anaconda测试的python
  7. 安装Rapids AI(最好在安装数百个“ conda”其他软件包以尝试与Rapids依赖关系图自洽一致之前立即安装此软件包-您始终可以通过安装其他python软件包点子或conda。)

完成此操作后,如果启动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。

答案 1 :(得分:1)

RAPIDS 最近发布的 a blog 概述了该程序。

步骤如下:

  1. 注册 Windows Insider,选择加入 Dev Channel,并更新到合适的 Windows Insider 版本 (>20145)
  2. Install Nvidia's preview driver for WSL2
  3. 安装 WSL2
  4. 验证 CUDA + WSL2 安装
  5. 安装 RAPIDS(通过 conda)

正如博客所建议的,首先按照 Nvidia's documented procedure 安装和配置带有 CUDA 的 WSL2。这应该涵盖步骤 (1-4)。另请参阅 RAPIDS 博客,其中包含有用的补充材料(尤其是在遇到问题时)。

安装和验证后,RAPIDS 可以通过 WSL2 命令行安装

conda create -n rapids-0.17 -c rapidsai -c nvidia -c conda-forge \
-c defaults rapids-blazing=0.17 python=3.7 cudatoolkit=10.1

注意:此命令将创建一个名为 rapids-0.17 的 conda 环境并在那里安装依赖项。要使用该环境,您必须使用 conda activate rapids-0.17 激活它。

如果您想在 RAPIDS 中使用替代版本的 Python,只需将此命令替换为 RAPIDS release selector 的一个输出。

答案 2 :(得分:0)

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

如果可以将GPU分配给Windows主机上的Linux VM,则唯一的选择是。这种可能但足够复杂,因此双重引导是一个更好的解决方案。