分享到: 分享到QQ  分享到Twitter

作者: BigLoser    访问次数: 1065 创建时间: 2020-06-21 21:19:32 更新时间: 2024-04-20 20:25:51

在微软最新发布的 Windows Insider 预览版本中,WSL2 获得了 GPU 计算支持。这意味着 Linux 二进制文件可以利用 GPU 资源,在 WSL 中进行机器学习、AI 开发或是数据科学等工作。

 

微软在今年五月份的 Build 2020 大会上宣布了 WSL 对 GPU 计算的支持,对这项功能的需求在社区中一直拥有很高的呼声。当时,微软表示,最初的 GPU 计算功能将提供两种方案:

 

  • NVIDIA CUDA (CUDA 是跨平台的 API,可以通过 Windows 上的 WDDM GPU 抽象或 Linux 上的 NVIDIA GPU 抽象与 GPU 通信)
  • DirectML(直接机器学习,机器学习的低级 API)

 

此次更新包括对 NVIDIA CUDA 和 DirectML 的支持。CUDA 的初始预览包括对现有机器学习工具、库和流行框架(包括 PyTorch 和 TensorFlow)的支持,以及本机 Linux 环境中可用的所有 Docker 和 NVIDIA Container Toolkit 支持,允许构建在 Linux 上运行的容器化 GPU 工作负载在 WSL 2 中按原样运行,让开发者可以在本地 Windows 计算机上进行开发和实验。它的面向目标是专业人士。

 

DirectML 则更加针对初学者,能够让使用者跨 AMD、Intel 和 NVIDIA GPU,在 Windows 硬件上实现加速,体验机器学习的工作负载。这次预览版本发布了带有 DirectML 后端的 TensorFlow 预览包,初学者可以从 TensorFlow 教程模型或微软提供的示例开始学习。

微软表示,计划在接下来的几个月中开源与 DirectML 一起使用的 TensorFlow 代码扩展库。

年度最有价值文章

季度最有价值文章

月度最有价值文章

投票统计

是否原创: 0 %

0 % Complete (success)

是否有价值: 0 %

0% Complete

是否有素质: 0 %

0% Complete (warning)

是否合法: 0 %

0% Complete

   群组工具

   外部链接