让tf支持gpu

Installing Bazel on Fedora and CentOS

安装bazel

TensorFlow 1.13.1: ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory · Issue #26209 · tensorflow/tensorflow

tf 1.13的依赖版本,cuda必须是10以上

How to get the cuda version?

cuda版本查看方式

安装支持gpu的tf的最简单的方式

就是pip install tensorflow-gpu

注意,cuda版本支持情况,install的时候指定版本。

    原文作者:王王
    原文地址: https://zhuanlan.zhihu.com/p/61534021
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞