Install

TensorLayer3 has some prerequisites that need to be installed first, including TensorFlow , MindSpore, PaddlePaddle,numpy and matplotlib.For GPU support CUDA and cuDNN are required.

# for last stable version
pip install --upgrade tensorlayer3

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

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

Alternatively, you can install the latest or development version by directly pulling from OpenI:

pip3 install git+https://git.openi.org.cn/TensorLayer/tensorlayer3.0.git

# 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

# 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

Contribute

Please read the Contributor Guideline before submitting your PRs.

Cite

If you find this project useful, we would be grateful if you cite the TensorLayer papers.

@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}
}

License

TensorLayer3 is released under the Apache 2.0 license.