Skip to content

Commit

Permalink
dev(narugo): add new init script
Browse files Browse the repository at this point in the history
  • Loading branch information
narugo1992 committed Dec 29, 2024
1 parent 71353ed commit eb42423
Show file tree
Hide file tree
Showing 3 changed files with 25 additions and 0 deletions.
2 changes: 2 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,8 @@ RUN python -V && pip -V
RUN pip install -U hfutils && hfutils whoami
ADD pyskeb.sh /pyskeb.sh
RUN chmod +x /pyskeb.sh
ADD yolov8.sh /yolov8.sh
RUN chmod +x /yolov8.sh

RUN ls -al /data

Expand Down
Empty file modified pyskeb.sh
100644 → 100755
Empty file.
23 changes: 23 additions & 0 deletions yolov8.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
YOLOV8_DIR=/data/yolov8

git clone https://github.com/deepghs/yolov8.git ${YOLOV8_DIR}
cd ${YOLOV8_DIR}

conda create -n yolov8 python=3.10 cudnn cuda-cudart libcufft -y
source activate yolov8
pip install -r requirements.txt
pip install -r requirements-onnx.txt
pip install -U dghs-imgutils onnxruntime onnxslim
conda info
python --version
pip list

conda deactivate
conda create -n yolov8raw python=3.10 cudnn cuda-cudart libcufft -y
source activate yolov8raw
pip install -r requirements-raw.txt
pip install -r requirements-onnx.txt
pip install -U dghs-imgutils onnxruntime onnxslim
conda info
python --version
pip list

0 comments on commit eb42423

Please sign in to comment.