Ubuntu16.04版本搭建Tensorflow环境并启用GPU支持

本文详细记录了在Ubuntu16.04上配置Tensorflow-GPU环境的过程,包括安装Tensorflow、CUDA和cuDNN的步骤,以及遇到的常见问题和解决方法,旨在为读者提供一个参考指南。
摘要由CSDN通过智能技术生成

Ubuntu16.04版本搭建Tensorflow环境并启用GPU支持

在配置Tensorflow并开启GPU支持的过程中遇到了一些问题,写这篇文章记录一下

环境搭建的基本流程

  • 在ubuntu上安装支持GPU版本的Tensorflow,特别需要注意的是针对Ubuntu16.04系统,Tensorflow中文站上的版本已经有些过时,在开启GPU支持时(安装Cuda合cuDNN)会出现问题,所以最好直接按照Github上面的instruction来安装,地址: Tensorflow-Github
  • 简单翻译一下安装步骤:

    1. TF需要python2.7或者python3.3+的支持,并且GPU必须时Cuda-capable的,可以在这里查看自己的n卡有没有被Cuda支持GPU
    2. 使用pip命令进行安装

      $ pip install tensorflow-gpu

      如果上面这种安装方式速度太慢,可以采用如下的方式:

      • 针对python2.7:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值