Linux/Ubuntu下安装配置Tensorflow教程

发布于 2020-9-22 11:02
浏览
0收藏

#摘要


之前一直用的caffe,但是最近突然有新需求,caffe对某些方面支持的不是很人性化,所以只能试试Tensorflow,毕竟Google出品必属精品,本文主要记录Ubuntu下安装Tensorflow的方法,如果是Windows或者mac环境请参考官方文档。

 

本文使用最简单的 pip 安装. 使用 pip 安装的时候要确保你的 pip 已经存在于你的电脑中. 如果还没有安装 pip. 你可以在 Terminal 窗口中运行这个, 升级必要的组件:

 

$ sudo apt-get install python-pip python-dev


##GPU 版

 

我电脑上之前已经装过Cuda和cudnn,之前没有装的同学请参考这里安装Cuda和cudnn的部分。

然后确保你的 linux 上 pip 是可用的, 接着我们可以直接通过pip 安装:

 sudo apt-get install python-pip python-dev   # for Python 2.7
 sudo apt-get install python3-pip python3-dev # for Python 3.n

然后选择你想要cpu 或者 gpu 版本.

# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

$ sudo -H pip install --upgrade $TF_BINARY_URL

 

##验证安装成功


在终端下执行:

python
import tensorflow


看到类似下图就ok:Linux/Ubuntu下安装配置Tensorflow教程-开源基础软件社区

 

作者:ai-exception

来源:CSDN

分类
收藏
回复
举报
回复
添加资源
添加资源将有机会获得更多曝光,你也可以直接关联已上传资源 去关联
    相关推荐