美文网首页
WSL下的Ubuntu使用带GPU 的TensorFlow

WSL下的Ubuntu使用带GPU 的TensorFlow

作者: 青云直上 | 来源:发表于2022-05-10 17:53 被阅读0次

微软为了改变办公操作系统的形象也是拼了，出了WSL支持原生虚拟linux系统，还为WSL做了多个优化，其中就包括提供TensorFlow在WSL下的GPU支持。

环境要求

  • Windows11 或者 Windows 10 （21H2版本以后）

  • WSL下的Linux，如Ubuntu或Debian

  • 只支持 Python 3.5, 3.6, 3.7

安装最新的显卡驱动

AMD

Download and install AMD's driver

Intel

Download and install Intel's driver to use with DirectML from their website.

NVIDIA

Download and install NVIDIA's driver to use with DirectML from their website. For more information, see NVIDIA's GPU in Windows Subsystem for Linux (WSL) page.

安装

pip install tensorflow-directml

官方网站说只支持Tensorflow1.15，实测安装的已经到2.6了。应该是网站还没更新。

参考链接

TensorFlow with DirectML on WSL | Microsoft Docs

相关文章

网友评论

      本文标题：WSL下的Ubuntu使用带GPU 的TensorFlow

      本文链接：https://www.haomeiwen.com/subject/ttzlurtx.html

      延伸阅读

      深度阅读

      • 您也可以注册成为美文阅读网的作者，发表您的原创作品、分享您的心情！

      栏目导航

      热点阅读

      关于我们|服务条款|联系我们|WSL下的Ubuntu使用带GPU 的TensorFlow|投稿指南|网站地图|RSS订阅|排版工具|手机版

      提供经典美文摘抄,优美散文欣赏,现代诗歌精选,短篇小说,心情随笔,表白情书范文,故事会在线阅读欣赏

      Copyright © 2014-2023 Haomeiwen.com All Rights Reserved. 好美文阅读网 版权所有

      备案信息：桂公网安备 45052102000051号 · 桂ICP备13007215号-3

      本站所收录作品、热点评论等信息部分来源互联网，目的只是为了系统归纳学习和传递资讯

      所有作品版权归原创作者所有，与本站立场无关，如不慎侵犯了你的权益，请联系我们告知，我们将做删除处理！