搭建tensorflow-gpu环境

本文详细介绍了在Windows10系统上,使用NVIDIA 1050Ti显卡,从驱动安装到CUDA、cuDNN配置,再到Anaconda虚拟环境中TensorFlow-GPU版本的完整设置流程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1.硬件
Windows10, 1050Ti

2.软件
NIVID Display Driver431.70
CUDA 9.0
cuDNN 7.0.4
Anaconda 中自带python3.7

3.步骤
3.1 驱动
下载网站:https://www.geforce.cn/drivers
可以根据自己电脑的配置安装合适的驱动

3.2 CUDA
下载网站:https://developer.nvidia.com/cuda-toolkit-archiveyu
推荐:这里最好默认安装,这样会自动添加到系统环境中
遇到的问题:在安装的过程中有遇到说版本不兼容的问题
解决的办法:是在继续安装→自定义安装→然后只选择第一个安装,剩下的两个不安装
具体可见:https://blog.youkuaiyun.com/qq_29631521/article/details/85266621
检查成功与否:在运行框中输入nrcc --version 输出:版本号

3.3 下载cuDNN
下载网站:https://developer.nvidia.com/rdp/cudnn-archive 要注册,不要慌,就是注册。
之后把cuDNN压缩包解压后,把bin,include,lib里面的文件(共三个)分别拷贝到CUDA\v9.0\下的bin,include,lib文件里
原文链接:https://blog.youkuaiyun.com/u012400305/article/details/84590259

3.4用Anaconda创造虚拟环境python3.6
https://blog.youkuaiyun.com/cyglight8947/article/details/82867996
在虚拟环境下用pip安装tensorflow-gpu,注意一定要安装与CUDA想匹配的tensorflow的版本
具体为:
在这里插入图片描述
3.5测试及配置
新建一个项目→将Anaconda下的envs的python.exe添加为解释器
https://blog.youkuaiyun.com/qq_29631521/article/details/85266621
在这里插入图片描述
附加:
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值