Sign inSign up

beomi/transformers-pytorch-gpu

By beomi

ā€¢Updated almost 6 years ago

Jupyter & etc lib added from huggingface/transformers-pytorch-gpu

Image
0

1.3K

beomi/transformers-pytorch-gpu repository overview

This Docker image is managed in this repository: https://github.com/Beomi/transformers-pytorch-gpuā 

ā transformers-pytorch-gpu

šŸ’” Docker setup for Huggingface šŸ¤— Transformers + GPU + Jupyter notebook + OhMyZsh

ā Features

  • Base Docker image: NVIDIA PyTorch 20.08 (1.7.0)
    • Built-in PyTorch 1.7 & CUDA
  • Added pip packages:
    • jupyter
    • pytorch-lightning
    • neptune-client
    • transformers
    • emoji
    • soynlp
    • konlpy
  • Custom uid support
  • Zsh & OhMyZsh support
  • Default LANG to en_US.UTF-8

ā How to use

šŸ’” Do this first! šŸ’”

id -u && id -g

If your result is like below:

$ id -u && id -g
1000
1000

GOTO -> Pull from dockerhubā 

Else:

(example)

$ id -u && id -g
1004
1004

GOTO -> Build from scratchā 

ā Pull from dockerhub
git clone https://github.com/Beomi/transformers-pytorch-gpu
cd transformers-pytorch-gpu
docker pull beomi/transformers-pytorch-gpu:latest
./run.sh

Now open http://your-server-ip:42000ā 

Jupyter Notebook is waiting for you :)

ā Build from scratch
git clone https://github.com/Beomi/transformers-pytorch-gpu
cd transformers-pytorch-gpu 
./build.sh
./run.sh

Now open http://your-server-ip:42000ā 

Jupyter Notebook is waiting for you :)

Tag summary

Content type

Image

Digest

Size

4.6 GB

Last updated

about 6 years ago

docker pull beomi/transformers-pytorch-gpu