前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >简单易学:Tensorflow CUDA 版本安装

简单易学:Tensorflow CUDA 版本安装

作者头像
一个有趣的灵魂W
发布2020-10-30 11:26:03
9560
发布2020-10-30 11:26:03
举报

看了那么多Tensorflow(GPU)安装教程,一个字,迷糊!

来一期简版教学吧:

1、想用GPU,就得确定你的电脑支不支持CUDA:

这里各位可以直接百度你GPU的型号,不展开了。

2、打开英伟达官网,选择对应版本:

代码语言:javascript
复制
https://tensorflow.google.cn/install/source_windows?hl=zh_cn

(包括python版本、编译器、MSVC、cuDNN、CUDA)

2.1

看好版本后,就照着下就完了~ Python(3.7)我使用Anaconda,具体参照以下:

python都拿你没办法,历史版本Anaconda下载

注:为什么选择3.7?

你敢信?新版的Python居然因为这个原因被束缚住了。

2.2

MSVC--该处我选择2019版本

代码语言:javascript
复制
https://visualstudio.microsoft.com/zh-hans/vs/

接着安装,选择单个组件:

安装就完了。

2.3 cuDNN,

选择对应7.4.(实际上cuDNN中7.4没有对应的CUDA10.1,所以我们选择了7.5)。cuDNN需要先注册:

代码语言:javascript
复制
https://developer.nvidia.com/rdp/cudnn-archive

注册后选择:

下载好后,解压,将bin添加到环境变量中。(环境变量如何添加,可以百度一下)

代码语言:javascript
复制
https://baijiahao.baidu.com/s?id=1652500747054512497&wfr=spider&for=pc

路径:

path

2.4 CUDA

CUDA-下载CUDA10.1,如果你电脑CUDA版本不匹配,可以先卸载CUDA或者英伟达软件,继而安装需要的版本。下载地址:

代码语言:javascript
复制
https://developer.nvidia.com/cuda-toolkit-archive

3、安装Anaconda

你可以直接从官网下,也可以选择下载历史版本:

历史版本Anaconda下载 F君的小尾巴,公众号:一个有趣的灵魂Wpython都拿你没办法,历史版本Anaconda下载

4、安装Tensorflow(GPU)

建议官网下载。

代码语言:javascript
复制
https://tensorflow.google.cn/install/pip?hl=zh_cn

(下载器下载就得了)接着安装CUDA。

5、最后pip

给小白提个醒,下载好的whl文件安置在D盘根目录下,然后写入

代码语言:javascript
复制
pip install tensorflow_gpu-2.3.0-cp37-cp37m-win_amd64.whl

接着测试,显示是True就是装好了。。。(收工,关注回“tfgpu”有word版下载)

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-10-28,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 一个有趣的灵魂W 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档