2019年8月 最新 Windows10+GPU版本tensorflow/ keras / pytorch+cuda+cudnn安装 搭建深度学习环境

安装思路

一般的教程都是先装cuda再装tensorflow/pytrch/keras,但是这样很容易造成版本不匹配!!
先确定tensorflow版本,由tf确定需要的cuda版本及cudnn版本,在去查看显卡驱动支不支持该cuda,不支持的话升级驱动。

由tensorfow确定cudn版本的方法:
安装tensorflow-gpu,安装完成后,import tensorflow,会报错,错误会提示,需要安装什么版本的cuda,如图:
在这里插入图片描述
截止2019-08-15,推荐直接pip install tensorflow安装tf1.14版本,配cuda10.0
安装默认已经装好了anaconda,使用anaconda prompt执行命令。

1、安装tensorflow

官网推荐创建虚拟环境:conda create -n tensorflow
进入虚拟环境:activate tensorflow
pip install tensorflow-gpu
该命令是默认安装最新版本的tf,2019-08-15时,安装的是1.14版本,对应cuda是10.0。

2、查看驱动支持版本

先打开nVidia控制面板-帮助-系统信息-组件里面能看到支持的最新的cudn

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值