win10+tensorflow-gpu+1050ti(终于安装成功了T﹏T)

前言 

可能很多人一开始跟我一样通过查看官网(CUDA-GPU支持关系)发现1050ti没有对应的CUDA版本,这也是我最开始放弃在自己笔记本上安装TF-gpu版的主要原因。直到疫情原因回不去学校实验室......然后踩坑之路就这样开始了╯﹏╰

其实写这篇文章的主要目的还是对自己艰难摸索的可行方法的一个记录,虽然未必在其他小伙伴的机器上一定可行(因为我就是这样子了,总会遇到别人帖子里遇不到的问题 o(╥﹏╥)o ),但相信多少可以给大家带来一点帮助。

推荐版本搭配

VS2015 + CUDA9 + Cudnn7.3.1 + Python3.6 + tensorflow-gpu1.7.0

网盘链接:https://pan.baidu.com/s/1ABlb0IK9Kt8HXj6wQG8n4Q 提取码:fv70

安装过程中需要用到的内容已经整理在网盘中方便大家取用,只不过整个压缩包有2.1G,百度网盘下载的话Emm...可能需要点耐心。如果实在等不及也可以去官网下载,而且是不用科学上网的速度也还挺快,只不过有些版本包可能不太好找。

VS2015社区版:链接:https://pan.baidu.com/s/149GICOdkdv9dpiWhH_YHSg  提取码:x3w9

Anaconda3-5.1.0:链接:https://pan.baidu.com/s/1bO_NWwMmQPiW0XTPNS_j8w 提取码:nl2a 

CUDA9.0.176_win10:https://developer.nvidia.com/cuda-90-download-archive

Cudnn7.3.1:https://developer.nvidia.com/rdp/cudnn-archive

安装

  • VS2015

文件解压后运行vs_community.exe,功能选择中只选择C++即可,然后就一路Next直到Finish就好啦(。◕ˇ∀ˇ◕)安装完成后无需启动,总之放那就OK的。

  • Anaconda

之所以选择该版本Anaconda(当然其他版本也有而可能是合适的)主要是因为自己之前版本的conda是默认Python3.7的,结果在创建Py3.6的虚拟环境时一直处于安装状态(小斜杠一直在转...),这貌似国内镜像源现在无法使用?但如果不用镜像源的话使用以下指令创建虚拟环境:

conda create -n tensorflow-gpu python=3.6

这时又会发现Python3.6的package下载超慢,而且最终还会出错 (ー`´ー)。所以说,我最后还是干脆重新安装了以上版本的Anaconda,然后在Anaconda Prompt中输入以上指令创建名称为tensorflow-gpu(名称当然是可以随便取的哦)的虚拟环境用作自己GPU版TF环境的搭建(先不要着急输指令,在这之前先把CUDA和Cudnn安装好)。

  • CUDA+Cudnn

运行下载好的CUDAxxx.exe,然后同样一路Next就好了(我是真的只点next,中间任何框框都没有✔)。因为我在这一步倒是很顺利地完成了没有遇到什么坑,但看很多人的帖子这个地方还是挺容易出现问题的,总之祝各位好运哦(๑•̀ㅁ•́ฅ)

Cudnn也比较简单,解压后放入CUDA安装位置的对应文件夹下就好了,最后再进行CUDA环境变量的配置。这个网上的参考文章很多,而且一般都没什么问题。比如:https://zhuanlan.zhihu.com/p/94220564?utm_source=wechat_session 。只不过可能需要注意一下环境变量配置完之后或许需要重启一下电脑才会生效(我就是因为这个问题花了好多时间,明明环境配置的路径下有某个DLL文件,但就显示缺少该文件<(`^´)>)

  • TensorFlow-GPU

首先激活创建好的虚拟环境:

activate tensorflow-gpu

升级pip到最新版,防止稍后的安装时出现错误:

python -m pip install --upgrade pip

安装tensorflow1.7.0及相应依赖包:

pip install --ignore-installed --upgrade tensorflow-gpu==1.7.0

至此,TF的GPU版本在1050ti上基本就安装完成了,可以输入以下代码试一下:

import tensorflow as tf
hello = tf.constant(‘Hello World!’)
sess = tf.Session()
print(sess.run(hello))

# 输出:b' Hello World!'

若遇到“FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecat”这个问题,可以参考该解决方案,亲测有效哦(・∀・)~~

看了一眼时间都凌晨5点了●﹏●再不睡赶不上午饭了。。。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值