基于Windows下安装TensorFlow-GPU和配置CUDA以及CUDDN

一、电脑是否支持GPU版本


查看方法:
windows + R在cmd中输入dxdiag
在这里插入图片描述
可以看到我的显2是NVIDIA生产的GeForce系列的

这个时候就可以去NVIDIA官网去下载自己芯片对应的版本
我的是GTX 1050 所以找相对应的
NVIDIA官网

在这里插入图片描述


二、安装TensorFlow-gpu

首先我的Python版本是Python 3.7的,TensorFlow是可以支持的。

安装TensorFlow-gpu命令
pip install tensorflow-gpu -i 加速源


三、安装CUDA和CUDDN

TensorFlow和cuda以及cuDDN也有版本对应关系的,所以在安装的时候先查看自己的TensorFlow的版本。
在命令行中输入:pip list
在这里插入图片描述
可以看到我的TensorFlow版本是2.0.0

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值