2022-12-10 21:18:03 +00:00
|
|
|
from nvidia/cuda:11.7.0-devel-ubuntu20.04
|
2023-05-27 20:50:47 +00:00
|
|
|
from python:3.11
|
2022-12-10 21:18:03 +00:00
|
|
|
|
|
|
|
env DEBIAN_FRONTEND=noninteractive
|
|
|
|
|
2022-12-24 13:39:40 +00:00
|
|
|
run apt-get update && \
|
|
|
|
apt-get install -y ffmpeg libsm6 libxext6
|
|
|
|
|
2022-12-10 21:18:03 +00:00
|
|
|
workdir /skynet
|
|
|
|
|
2022-12-17 14:39:42 +00:00
|
|
|
copy requirements.cuda* ./
|
2022-12-10 21:18:03 +00:00
|
|
|
|
|
|
|
run pip install -U pip ninja
|
2023-01-16 03:25:16 +00:00
|
|
|
run pip install -v -r requirements.cuda.0.txt
|
2022-12-10 21:18:03 +00:00
|
|
|
run pip install -v -r requirements.cuda.1.txt
|
2023-01-16 03:25:16 +00:00
|
|
|
run pip install -v -r requirements.cuda.2.txt
|
2022-12-10 21:18:03 +00:00
|
|
|
|
2022-12-17 14:39:42 +00:00
|
|
|
copy requirements.txt requirements.txt
|
|
|
|
copy pytest.ini pytest.ini
|
|
|
|
copy setup.py setup.py
|
|
|
|
copy skynet skynet
|
|
|
|
|
2023-05-27 20:50:47 +00:00
|
|
|
run pip install -e . -r requirements.txt
|
2022-12-10 21:18:03 +00:00
|
|
|
|
2022-12-17 14:39:42 +00:00
|
|
|
env PYTORCH_CUDA_ALLOC_CONF max_split_size_mb:128
|
2022-12-10 21:18:03 +00:00
|
|
|
env NVIDIA_VISIBLE_DEVICES=all
|
|
|
|
env HF_HOME /hf_home
|
|
|
|
|
2022-12-17 14:39:42 +00:00
|
|
|
copy tests tests
|