安装tensorflow-gpu

安装前,先说明自己学习深度学习tensorflow框架,自己买了GTX1070显卡,利用起来。

一、安装python3.7(我选择的,任意)

二、安装CUDA、cudnn

1、从NVIDIA官网(https://developer.nvidia.com/cuda-downloads)下载自己需要的CUDA版本(我选择的10.0),默认一直安装,注意勾选添加到路径。

2、从NVIDIA官网(https://developer.nvidia.com/cudnn)下载自己需要的cudnn版本(要与CUDA版本相匹配),下载下来一个压缩包,解压后,将文件夹里所有文件粘贴到CUDA安装路径下。

三、安装tensorflow-gpu

在线安装方式很多conda、pip都可以

离线安装:强烈建议!

https://pypi.tuna.tsinghua.edu.cn/simple/  上这个网站找资源,下载好.whl文件。(也就是轮子)

pip install c:\...\tensorflow_gpu-xxxx.whl

四、安装完测试

遇到问题:

1、CUDA与tensorflow版本不匹配,一开始我用的10.1,改成10.0.

tensorflow官网可以看对应版本:https://tensorflow.google.cn/install/source_windows

2、我使用了anaconda,里面的虚拟环境配置学习了。

备注:

https://www.bilibili.com/video/av52282208?from=search&seid=2657292146605146276这个教程在线教安装

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值