tensorflow GPU (1080ti) + win10安装笔记

最近一周安装tensorflow-gpu 踩坑无数,昨天安装成功后,发现如果按正确流程,又是如此简单顺利(3小时基本可以搞定)。记录下来,供大家参考,少踩本人的坑。。。


一、GPU的硬件选择

1080ti是性价比非常高的GPU,用MNIST测试,GPU版tensorflow程序比cpu版至少快20倍以上。我选择的是索泰的品牌,感觉料足(3斤多,沉甸甸),电话客服的态度和专业性也很不错。 但在选GPU时注意:

1、主机电源的是否支持。1080以上显卡是8+6pin(或8+8),需要从电源的两个接口供电(一个8转出8+6,不稳定)。对电源的稳定性和功率要求较高。1080ti额定功率250W,因此主机电源功率越大越好,700W以上较稳妥。电源如果达不到要求,GPU运行中可能会导致电脑重启。

2、显卡插槽和机箱空间是否足够。1080ti长度在30厘米多点,要注意显卡插槽延伸线上是否有凸起的元器件影响插入。显卡厚度占满两个pci槽,索泰背板还有小风扇,装上后就需要3个槽的宽度。

3、如果采用多GPU,所有GPU必须都是相同型号。


二、Windows下所需的软件与版本选择

经历这次装tensorflow-gpu出现的各种问题后,我觉得搭配好各组件版本是关键,以下是本人比较顺利安装的版本,一定还有很多其他组合,供大家参考。

版本搭配:Win10+VS2015+anaconda5.0+python3.5+CUDA8.0.61+cuDNN6.0+tensorflow-gpu1.4

接下来我写一下相关软件安装时特别注意的地方(哥的踩坑经验。。。)

对了,安装时最好关闭360这些杀毒程序,使用Administrader帐户。

1、win10 没什么说的,主要因为CUDA和cuDNN对win7的版本少。至少cuDNN6的win7版我没找到。还有网上经验说CUDA在win7下应使用8.0.44版等,既不好找,也不是主流版本,可能会和其他组件有冲突,所以建议使用win10。 另外一个细节是,win10要在线升级到较新版本,不然1080ti的驱动无法安装。好像是GPU型号越高级,对Win10版本需要越新。

2、安装VS2015。只是因为CUDA安装需要VS支持,建议使用2015。VS2013在win10下好像需要使用兼容模式安装,可能比较麻烦。

3、安装CUDA8.0.61。 第一次装CUDA可能比较晕, 新人注意以下步骤(已经安过的要全部删干净,最好连驱动也卸了重装一次,删干净相应的注册表项):

a、下载cuda8.0.61版本(NVIDIA官网是9.0版,如有需要可私信)

b、安装时仔细看英文提示,不要改路径(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\)。首先会进行系统兼容性检查,像1080ti以上这种较高端的显卡,会提示cuda中的驱动比GPU现有的驱动版本低,建议使用现有驱动。如果出现这个提示,不用担心,在“下一步”选择安装模式时,把“默认”改为自定义。 然后下一步的三个钩选框中,只选择安装CUDA,不要其他两个(好像是驱动和物理加速程序)。 其次需要注意,在最后会提示已经安装和没有安装的VS版本。如果已经安装了VS2015和.Net framework4.5以上的,不用理会了。

c、需要手动修改path。右键打开我的电脑-->属性-->高级系统设置-->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V8_0两个环境变量。接下来,还要手动添加以下几个环境变量:

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值