安装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不工作的问题。常见的问题如下:
4. h5版本问题
此问题的解决方法是重新安装h5:
pip uninstall h5py
pip install h5py
5. cudatoolkit与显卡驱动不匹配问题
此问题的解决方法是升级显卡驱动。这种的问题产生的一个原因就是电脑是以前生产的或者设计定型的。显卡的驱动不是最新版。一般而言,nvidia的显卡都是支持tensorflow gpu的最新版本的。出现问题的原因就是显卡驱动是旧版本。
安装显卡驱动最好的方法是使用手动搜索去找安装包。步骤见我的上一篇微博。
7. 下载速度慢
建议使用国内的镜像网站

发布了92 篇原创文章 · 获赞 30 · 访问量 2万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览