lichen/ieemoo-ai-zhanting
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

3588适配

Browse Source
This commit is contained in:
李晨
2024-10-17 19:14:27 +08:00
parent 8475980895
commit a95db2a8fb
9 changed files with 78 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@ -1,5 +1,5 @@
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
FROM nvcr.io/nvidia/pytorch:21.03-py3
FROM nvcr.io/nvidia/pytorch:23.06-py3
# Install linux packages
RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
@ -20,6 +20,10 @@ COPY . /usr/src/app
# Set environment variables
ENV HOME=/usr/src/app
EXPOSE 8000
CMD ["python", "ieemoo-ai-zhanting.py"]
# --------------------------------------------------- Extras Below ---------------------------------------------------