linux下Anaconda安装tensorflow-gpu其实很简单

6人阅读 评论(0) 收藏 举报
分类:

之前网上看了很多教程,走了很多弯路,感觉有必要记一下,cuda、cudnn也都不需要自己装,conda会在装tensorflow-gpu时帮你搞定,省了很多事。步骤如下:

一、安装驱动

https://blog.csdn.net/weixin_39954229/article/details/79939599

二、安装Anconda

1、https://www.anaconda.com/download/#linux 下载

2、cd到文件下载的位置输入如下命令安装

$ bash Anaconda3-5.1.0-Linux-x86_64.sh3、重启终端

4、创建一个tensorflow-gpu环境

$ conda create -n tf python=3.6

三、安装tensorflow

1、激活环境:source activate tf

2、在终端输入以下命令搜索当前可用的tensorflow版本:

$ anaconda search -t conda tensorflow

3、选择一个较新的CPU或GPU版本,如anaconda/tensorflow-gpu 1.4.1版本,

输入如下命令查看详细信息:

$ anaconda search -t conda tensorflow-gpu
然后得到如下信息:

Using Anaconda API: https://api.anaconda.org
Name:    tensorflow-gpu
Summary: TensorFlow is a machine learning library.
Access:  public
Package Types:  conda
Versions:
   + 1.0.1
   + 1.1.0
   + 1.2.1
   + 1.3.0
   + 1.4.1

To install this package with conda run:
     conda install --channel https://conda.anaconda.org/anaconda tensorflow-gpu
4、输入最后一行命令安装:
$ conda install --channel https://conda.anaconda.org/anaconda tensorflow-gpu

5、测试安装是否成功:

$ python3 
>> import tensorflow
查看评论
    个人资料
    等级:
    访问量: 684
    积分: 80
    排名: 155万+
    文章存档