欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

我的AI进坑过程 --- TensorFlow安装 (1)

程序员文章站 2022-06-12 16:59:34
...

电脑配置好(主要是显卡)的话可以安装cpu与gpu版本,不行话安装cpu

目录

1.python 安装

2.cuda安装

 3.更新pip ,不然可能会出错

4.TensorFlow安装


1.python 安装

   自行百度,很多教程

2.cuda安装

     详情可以看这里 cuda安装详情(侵权删)

 3.更新pip ,不然可能会出错

我的AI进坑过程 --- TensorFlow安装 (1)

    命令:python -m pip install --upgrade pip

    pip源地址:https://pypi.org/project/pip/

4.TensorFlow安装

(1)使用native pip 安装TensorFlow(本人不推荐,因为装的cpu和gpu版本会偏低,容易报错,烦死了)

         CPU版本:pip3 install --upgrade tensorflow

         GPU版本:pip3 install --upgrade tensorflow-gpu

 (2)下载 wheel 文件 来安装(推荐

        这里有大神编译好的CPU和GPUwheel文件,但是只能用在python3.5和3.6,如果各位大神们有其他版本的,欢迎分享

        进入下载好的目录然后 :pip install 文件名.whl就好了

     实在找不到,可以试试:把里面的版本换成自己想要的

cpu:

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.1.0-cp35-cp35m-win_amd64.whl

gpu:

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl 

坑:(1)TensorFlow GPU和cpu版本与python版本要对应:比如 tensorflow-1.8.0-cp36-cp36m-win_amd64   tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64 这个两个文件,cp36m表示python3.6版本的,要想安装1.8版本的TensorFlow

需要用3.6版本的python,第二个也是雷同,要想安装1.8版本的GPU需要用python3.6,如果版本不对就会报错

        (2)AttributeError: module 'tensorflow.python.platform.flags' has no attribute 'xxx',如果报这种类型的错误,建议用第whl方法升级一下TensorFlow

 

相关标签: AI