Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

refactor code using black formatter and add dockerhub push cmd, and update base image to version containing cudnn8 #13

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open

refactor code using black formatter and add dockerhub push cmd, and update base image to version containing cudnn8 #13

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
1b4a316
refactor code using black formatter
Tyrese-FullStackGenius Mar 20, 2024
7f05b6d
fix: base docker image
Tyrese-FullStackGenius Mar 20, 2024
0efab50
fix: clean temp files and cache
Tyrese-FullStackGenius Mar 20, 2024
f0188ec
fix: static import
FrostFlowerFairy Apr 23, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 1 addition & 8 deletions Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# Use an NVIDIA CUDA base image with Python 3
FROM nvidia/cuda:11.6.2-base-ubuntu20.04
FROM nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu20.04
Copy link
Collaborator

@jinmiaoluo jinmiaoluo May 20, 2024 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The Dockerfile in the current repository will lead to the following issues:

Client 583ffd78-2bfe-4e1e-bbbc-8efed00e1904 connected
2024-05-20T08:38:53.837582891Z Could not load library libcudnn_ops_infer.so.8. Error: libcudnn_ops_infer.so.8: cannot open shared object file: No such file or directory

The current commit can fix this issue.


# Set the working directory in the container
WORKDIR /usr/src/app
Expand All @@ -9,13 +9,6 @@ ENV DEBIAN_FRONTEND=noninteractive

# Install any needed packages
RUN apt-get update && apt-get install -y python3-pip libsndfile1 ffmpeg wget dpkg
RUN wget https://github.com/TyreseDev/nvidia-cucnn/releases/download/v0.0.1/cudnn-local-repo-ubuntu2004-8.9.7.29_1.0-1_amd64.deb && \
dpkg -i cudnn-local-repo-ubuntu2004-8.9.7.29_1.0-1_amd64.deb && \
cp /var/cudnn-local-repo-ubuntu2004-8.9.7.29/cudnn-*-keyring.gpg /usr/share/keyrings/ && \
apt-get update && apt-get -y install libcudnn8 libcudnn8-dev && \
rm -f cudnn-local-repo-ubuntu2004-8.9.7.29_1.0-1_amd64.deb && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*

# Copy the requirements.txt file
COPY requirements.txt ./
Expand Down