我正在尝试在 Windows 计算机上运行 Nvidia rapids,但没有任何运气。我已经为 windows 安装了 docker desktop 并下载了 rapids 图像。已安装 Cuda 10.0,但未安装 Nvidia-container-toolkit。我无法让它运行。有什么想法或指导吗?
2 回答
我不确定是否有人对原始问题给出了更明确的“更新”答案。在这一点上(2020 年 8 月),答案是“是的!”。您绝对可以在 Windows 10 上的 WSL2 中运行 RAPIDS,但需要满足以下几个条件:
要求
您必须在 Windows Subsystem for Linux version 2 (WSL2) 中使用 RAPIDS;
Windows 10 版本 2004(操作系统内部版本 202001.1000 或更高版本) 您必须注册才能获得 Windows Insider Preview 版本,特别是开发人员频道。这是 WSL2 VM 具有 GPU 访问权限所必需的。https://insider.windows.com/en-us/
CUDA SDK v11.1 中的 CUDA 版本 455.41 您必须使用特殊版本的 NVIDA CUDA 驱动程序(我正在使用),您必须通过从 NVIDIA 网站的特殊下载获得。您必须加入NVIDIA 开发人员计划才能访问该版本——然后搜索“WSL2 CUDA 驱动程序”,它应该会引导您找到它。
设置
- 安装开发者预览版的windows。确保单击“更新”中的复选框,该复选框也会安装其他推荐的更新。
- 从 NVIDIA Developer Program 安装 windows CUDA 驱动程序
- 通过启用“虚拟机平台”可选功能启用 WSL 2。您可以在此处找到更多步骤https://docs.microsoft.com/en-us/windows/wsl/install-win10
- 从 Windows 商店安装 WSL(Ubuntu-20.04 确认工作)
- 在 WSL 虚拟机上安装 python,用Anaconda测试
- 安装Rapids AI(最好在你有数百个其他“conda”包之前安装它,以尝试与 Rapids 依赖图自洽一致——你以后总是可以通过 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。
20 年 9 月 6 日更新: Wesley 写的答案在最新的带有 WSL2 的 Windows Insider Preview 中是准确的。我没有修改这个答案,而是对他进行了编辑。https://stackoverflow.com/a/59364773/6779504
不,因为它现在存在,RAPIDS 需要一个 Linux 主机。这是在 NVIDIA 最近的一次研讨会上提出的。还提到 RAPIDS 不适用于 WSL。它可能适用于 WSL 版本 2,但我没有尝试过,也不知道有人这样做。
如果您可以将 GPU 分配给 Windows 主机上的 Linux VM,则唯一的选择是。这可能但足够复杂,双引导是一个更好的解决方案。