打开APP
userphoto
未登录

开通VIP,畅享免费电子书等14项超值服

开通VIP
在windows上极简安装GPU版AI框架

原创 deepaiDL 实用AI客栈 2022-04-26 22:48

收录于合集

#windows1
#深度学习9
#人工智能10
#机器学习7
#计算机视觉2


对新手而言,安装深度学习框架通常需要大费周折,有没有省事简便的安装方法呢

如果我们想在windows系统上安装GPU版本的AI框架,比如GPU版本的tesnorflow,通常我们会看到类似下面的安装教程

官方版本

  1. 安装CUDA

  2. 安装cuDNN

  3. 配置环境变量

  4. 安装python环境

  5. 安装gpu版的tensorflow开发包

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

  • 下载的cuDNN时候需要注册,而且因为cuDNN文件在外网,下载速度很慢。

  • 比如不同版本的tensorflow和CUDA(cuDNN)的版本是有匹配关系的,那你可能会说我先查好他们的这个依赖关系,然后找好合适的版本再安装。

  • 比如对于windows系统而言,显卡驱动的版本决定了你能够安装的CUDA版本,有部分教程会忽略这一点。

  • ......

    假如你幸运地把上面的这些坑全部趟过了,那你还可能会遇到一个问题,就是如果你用pip去安装tensorflow-gpu,还是有可能会报各种神奇的错误。事实上,在windows系统上安装AI框架,最好的方法是用conda,而不是用原生的pip。总结了上面的这些常见的问题,下面我们开始介绍如何在windows上极简安装GPU版AI框架:

极简版本

  1. 安装Anaconda

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

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

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

  5. 结束啦

上面我们可以看到,这里我们使用了一个叫做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 -c pytorch

大家有没有觉得安装环境变得很简单了呢,欢迎评论区留言。

1

END

1

实用AI客栈

分享人工智能领域最新资讯,讲解各类实用的算法和模型。

20篇原创内容

公众号

确定

  • 不看此公众号

本站仅提供存储服务,所有内容均由用户发布,如发现有害或侵权内容,请点击举报
打开APP,阅读全文并永久保存 查看更多类似文章
猜你喜欢
类似文章
【热】打开小程序,算一算2024你的财运
win10下基于python(anaconda)安装gpu版本的TensorFlow以及kears深度学习框架
Windows +TensorFlow+Faster Rcnn 详细安装步骤(一)
Win10上安装Keras 和 TensorFlow(GPU版本)
TensorFlow在windows上安装与简单示例
超详细配置教程:用Windows电脑训练深度学习模型
用Windows电脑训练深度学习模型?超详细配置教程来了
更多类似文章 >>
生活服务
热点新闻
分享 收藏 导长图 关注 下载文章
绑定账号成功
后续可登录账号畅享VIP特权!
如果VIP功能使用有故障,
可点击这里联系客服!

联系客服