在windows上如何安装GPU版AI框架
小编给大家分享一下在windows上如何安装GPU版AI框架,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!
创新互联专注骨干网络服务器租用十余年,服务更有保障!服务器租用,服务器托管 成都服务器租用,成都服务器托管,骨干网络带宽,享受低延迟,高速访问。灵活、实现低成本的共享或公网数据中心高速带宽的专属高性能服务器。
官方版本
安装CUDA
安装cuDNN
配置环境变量
安装python环境
安装gpu版的tensorflow开发包
咋看上去好像不是很复杂,但是其中坑多到你怀疑人生。
下载的cuDNN时候需要注册,而且因为cuDNN文件在外网,下载速度很慢。
比如不同版本的tensorflow和CUDA(cuDNN)的版本是有匹配关系的,那你可能会说我先查好他们的这个依赖关系,然后找好合适的版本再安装。
比如对于windows系统而言,显卡驱动的版本决定了你能够安装的CUDA版本,有部分教程会忽略这一点。
......
假如你幸运地把上面的这些坑全部趟过了,那你还可能会遇到一个问题,就是如果你用pip去安装tensorflow-gpu,还是有可能会报各种神奇的错误。事实上,在windows系统上安装AI框架,最好的方法是用conda,而不是用原生的pip。总结了上面的这些常见的问题,下面我们开始介绍如何在windows上极简安装GPU版AI框架:
极简版本
安装Anaconda
新建conda虚拟环境(建议用python3.6,可换国内源)
安装显卡驱动(官网下载或者用驱动助手软件)
安装AI框架时同时安装cudatoolkit包
结束啦
上面我们可以看到,这里我们使用了一个叫做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框架”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注创新互联行业资讯频道!
分享标题:在windows上如何安装GPU版AI框架
当前网址:http://scyanting.com/article/ggjsjh.html