rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #103 from henk717/main 

Modern ROCm Docker
This commit is contained in:
henk717 2022-03-06 19:02:38 +01:00 committed by GitHub
parent 657de72ada 4332074c89
commit 932aabc2f3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
docker-rocm/Dockerfile
View File

@ -1,6 +1,6 @@
# This dockerfile is meant to serve as a rocm base image. It registers the debian rocm package repository, and FROM mambaorg/micromamba
# installs the rocm-dev package. WORKDIR /content/
COPY env.yml /home/micromamba/env.yml
FROM rocm/pytorch RUN micromamba install -y -n base -f /home/micromamba/env.yml
RUN conda install --all -y Flask-SocketIO USER root
RUN pip3 install git+https://github.com/finetuneanon/transformers@gpt-neo-localattention3 RUN apt update && apt install xorg libsqlite3-0 -y

1
play-rocm.sh
View File

@ -1,3 +1,4 @@
cd docker-rocm cd docker-rocm
xhost +local:docker xhost +local:docker
cp ../environments/rocm.yml env.yml
docker-compose run --service-ports koboldai bash -c "cd /content && python3 aiserver.py $*" docker-compose run --service-ports koboldai bash -c "cd /content && python3 aiserver.py $*"