JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-04-05 19:41:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated docker file and added script to instsall needed package post image

Browse Source
This commit is contained in:
Blake Mallory 2022-07-27 00:24:52 -04:00
parent f3461dac3a
commit 30fcd4f9ff
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Dockerfile
View File

@ -3,6 +3,9 @@ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
ARG DEBIAN_FRONTEND=noninteractive
ENV TZ=Europe/London
SHELL ["/bin/bash", "-c"]
RUN apt-get update && apt-get upgrade -y
RUN apt install git -y \
&& apt install python3 -y \
@ -10,8 +13,14 @@ RUN apt install git -y \
&& apt install git -y \
&& apt install python-is-python3 -y \
&& apt install python3-tk -y \
&& apt install ffmpeg libsm6 libxext6 -y
&& apt install ffmpeg libsm6 libxext6 -y \
&& apt install git -y
WORKDIR /workspace
RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 \
&& pip install opencv-python
RUN pip install SwissArmyTransformer>=0.2.9 icetk gifmaker

3
install_image_local_attention.sh Executable file
View File

@ -0,0 +1,3 @@
#!/bin/bash
pip install git+https://github.com/Sleepychord/Image-Local-Attention