skynet/Dockerfile

24 lines
387 B
Docker

from pytorch/pytorch:latest
run apt-get update && apt-get install -y git curl
run pip install --upgrade \
diffusers[torch] \
transformers \
huggingface_hub \
pyTelegramBotAPI \
pymongo \
pdbpp
run mkdir /scripts
run mkdir /outputs
run mkdir /inputs
env HF_HOME /hf_home
run mkdir /hf_home
workdir /scripts
env PYTORCH_CUDA_ALLOC_CONF max_split_size_mb:128