跳转到主要内容

为研究人员和工程师提供的高级Tensorflow深度学习库。

项目描述

TENSORLAYER-LOGO

Awesome Documentation-EN Documentation-CN Book-CN Downloads

PyPI PyPI-Prerelease Commits-Since Python TensorFlow

Travis Docker RTD-EN RTD-CN PyUP Docker-Pulls Code-Quality

JOIN-SLACK-LOGO

TensorLayer 是一个基于 TensorFlow 的深度学习和强化学习库,旨在为研究人员和工程师设计。它提供了一大批可定制的神经网络层/函数,这些是构建现实世界人工智能应用的关键。TensorLayer 被授予 2017 年 ACM 多媒体学会颁发的最佳开源软件奖。

设计特点

TensorLayer 是一个以简洁、灵活和高性能为设计理念的深度学习库。

  • 简洁性:TensorLayer 拥有高级层/模型抽象,易于学习。您可以在几分钟内通过大量的[示例](https://github.com/tensorlayer/awesome-tensorlayer)了解深度学习如何帮助您的 AI 任务。

  • 灵活性:TensorLayer API 透明且灵活,受 PyTorch 库的启发。与 Keras 抽象相比,TensorLayer 使构建和训练复杂 AI 模型变得更加容易。

  • 零成本抽象:尽管使用简单,但 TensorLayer 不需要您在 TensorFlow 的性能上做出任何妥协(请参阅以下基准部分以获取更多详细信息)。

TensorLayer 在 TensorFlow 包装器中占据一个独特的位置。其他包装器如 Keras 和 TFLearn 隐藏了 TensorFlow 的许多强大功能,并且对编写自定义 AI 模型的支持很少。受 PyTorch 启发,TensorLayer API 简单、灵活且 Pythonic,易于学习,同时足够灵活以应对复杂的 AI 任务。TensorLayer 拥有快速增长的社区。它已被全球的研究人员和工程师使用,包括北京大学、伦敦帝国理工学院、加州大学伯克利分校、卡内基梅隆大学、斯坦福大学以及 Google、Microsoft、阿里巴巴、腾讯、小米和彭博社等公司。

安装

TensorLayer 需要包括 TensorFlow、numpy 等在内的先决条件。对于 GPU 支持,需要 CUDA 和 cuDNN。安装 TensorLayer 最简单的方法是使用 Python 包索引(PyPI)

# for last stable version
pip install --upgrade tensorlayer

# for latest release candidate
pip install --upgrade --pre tensorlayer

# if you want to install the additional dependencies, you can also run
pip install --upgrade tensorlayer[all]              # all additional dependencies
pip install --upgrade tensorlayer[extra]            # only the `extra` dependencies
pip install --upgrade tensorlayer[contrib_loggers]  # only the `contrib_loggers` dependencies

或者,您可以直接从 github 拉取最新或开发版本进行安装

pip install https://github.com/tensorlayer/tensorlayer/archive/master.zip
# or
# pip install https://github.com/tensorlayer/tensorlayer/archive/<branch-name>.zip

使用 Docker - 一个现成的环境

基于官方 TensorFlow 容器TensorLayer 容器 已经构建

支持 CPU 的容器

# for CPU version and Python 2
docker pull tensorlayer/tensorlayer:latest
docker run -it --rm -p 8888:8888 -p 6006:6006 -e PASSWORD=JUPYTER_NB_PASSWORD tensorlayer/tensorlayer:latest

# for CPU version and Python 3
docker pull tensorlayer/tensorlayer:latest-py3
docker run -it --rm -p 8888:8888 -p 6006:6006 -e PASSWORD=JUPYTER_NB_PASSWORD tensorlayer/tensorlayer:latest-py3

支持 GPU 的容器

这些容器需要 NVIDIA-Docker 才能工作:[项目链接](https://github.com/NVIDIA/nvidia-docker)

# for GPU version and Python 2
docker pull tensorlayer/tensorlayer:latest-gpu
nvidia-docker run -it --rm -p 8888:88888 -p 6006:6006 -e PASSWORD=JUPYTER_NB_PASSWORD tensorlayer/tensorlayer:latest-gpu

# for GPU version and Python 3
docker pull tensorlayer/tensorlayer:latest-gpu-py3
nvidia-docker run -it --rm -p 8888:8888 -p 6006:6006 -e PASSWORD=JUPYTER_NB_PASSWORD tensorlayer/tensorlayer:latest-gpu-py3

贡献

在提交您的 PR 之前,请阅读贡献指南

引用

如果您发现这个项目很有用,我们将会非常感谢您引用 TensorLayer 论文。

@article{tensorlayer2017,
    author  = {Dong, Hao and Supratak, Akara and Mai, Luo and Liu, Fangde and Oehmichen, Axel and Yu, Simiao and Guo, Yike},
    journal = {ACM Multimedia},
    title   = {{TensorLayer: A Versatile Library for Efficient Deep Learning Development}},
    url     = {http://tensorlayer.org},
    year    = {2017}
}

@inproceedings{tensorlayer2021,
  title={Tensorlayer 3.0: A Deep Learning Library Compatible With Multiple Backends},
  author={Lai, Cheng and Han, Jiarong and Dong, Hao},
  booktitle={2021 IEEE International Conference on Multimedia \& Expo Workshops (ICMEW)},
  pages={1--3},
  year={2021},
  organization={IEEE}

许可证

TensorLayer 在 Apache 2.0 许可下发布。

项目详情


发布历史 发布通知 | RSS 源

下载文件

下载适合您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源分布

tensorlayer-2.2.5.tar.gz (277.1 kB 查看哈希值)

上传时间

构建分布

tensorlayer-2.2.5-py3-none-any.whl (381.2 kB 查看哈希值)

上传时间 Python 3

由以下组织支持

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误日志 StatusPage StatusPage 状态页面