在windows上如何安装GPU版AI框架

发布时间:2021-11-16 11:39:13 作者:小新
来源:亿速云 阅读:142

小编给大家分享一下在windows上如何安装GPU版AI框架,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!

官方版本

  1. 安装CUDA

  2. 安装cuDNN

  3. 配置环境变量

  4. 安装python环境

  5. 安装gpu版的tensorflow开发包

咋看上去好像不是很复杂,但是其中坑多到你怀疑人生。

极简版本

  1. 安装Anaconda

  2. 新建conda虚拟环境(建议用python3.6,可换国内源)

  3. 安装显卡驱动(官网下载或者用驱动助手软件)

  4. 安装AI框架时同时安装cudatoolkit包

  5. 结束啦在windows上如何安装GPU版AI框架

上面我们可以看到,这里我们使用了一个叫做cudatoolkit的包,相当于在conda的虚拟环境里面自动安装了CUDA。大多数时候cudatoolkit中的CUDA和cuDNN是一起安装的,这就省去了很多麻烦,比如说AI框架和CUDA的版本对应问题以及进行手动设置环境变量等等。

具体的安装脚本可参考如下命令:

安装tensorflow

conda create -n py36 python=3.6

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --set show_channel_urls yes

conda install cudatoolkit tensorflow-gpu

安装pytorch

conda create -n py36 python=3.6

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --set show_channel_urls yes

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

以上是“在windows上如何安装GPU版AI框架”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!

推荐阅读:
  1. 在 Windows 上安装 MongoDB
  2. GPU版的tensorflow在windows上的安装时的错误解决方案

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

windows gpu ai

上一篇:Intellij IDEA智能补全的方法有哪些

下一篇:spring oauth2 +springboot sso的案例分析

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》