安装Tensorflow-gpu的步骤

安装Tensorflow gpu版本还是比较繁琐的。经常会碰到各种问题。这个是简单的步骤总结:

  1. 安装anaconda
    此时要选好的是anacond和python的版本。
  2. 创建环境
    conda create -n tf
  3. 安装Tensorflow gpu
    conda install tensorflow-gpu
    此命令与pip install tensorflow-gpu 的区别是安装了多种依赖包,不用去单独安装了。强烈建议使用conda install tenforflow-gpu

一般而言,安装之后,都会碰到gpu不工作的问题。常见的问题如下:

  1. h5版本问题
    此问题的解决方法是重新安装h5:
    pip uninstall h5py
    pip install h5py
  2. cudatoolkit与显卡驱动不匹配问题
    此问题的解决方法是升级显卡驱动。这种的问题产生的一个原因就是电脑是以前生产的或者设计定型的。显卡的驱动不是最新版。nvidia的显卡都是支持tensorflow gpu的最新版本的。出现问题的原因就是显卡驱动是旧版本。
    安装显卡驱动最好的方法是使用手动搜索去找安装包。步骤见我的上一篇微博。
  3. 下载速度慢
    建议使用国内的镜像网站
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值