2022-12-06 20:50:04 +00:00
|
|
|
from pytorch/pytorch:latest
|
2022-12-04 23:39:43 +00:00
|
|
|
|
2022-12-06 12:29:41 +00:00
|
|
|
env DEBIAN_FRONTEND=noninteractive
|
|
|
|
|
|
|
|
|
|
run apt-get update && apt-get install -y git wget
|
2022-12-04 23:39:43 +00:00
|
|
|
|
2022-12-06 20:50:04 +00:00
|
|
|
run conda install xformers -c xformers/label/dev
|
|
|
|
|
|
2022-12-04 23:39:43 +00:00
|
|
|
run pip install --upgrade \
|
|
|
|
|
diffusers[torch] \
|
2022-12-06 20:50:04 +00:00
|
|
|
accelerate \
|
2022-12-04 23:39:43 +00:00
|
|
|
transformers \
|
|
|
|
|
huggingface_hub \
|
|
|
|
|
pyTelegramBotAPI \
|
|
|
|
|
pymongo \
|
2022-12-06 12:29:41 +00:00
|
|
|
scipy \
|
2022-12-04 23:39:43 +00:00
|
|
|
pdbpp
|
|
|
|
|
|
2022-12-06 20:50:04 +00:00
|
|
|
env NVIDIA_VISIBLE_DEVICES=all
|
|
|
|
|
|
2022-12-04 23:39:43 +00:00
|
|
|
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
|
