EuuIia commited on
Commit
0246318
·
verified ·
1 Parent(s): 1591227

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -61,6 +61,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
61
  ca-certificates libglib2.0-0 libgl1 \
62
  && apt-get clean && rm -rf /var/lib/apt/lists/*
63
 
 
 
 
64
  RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
65
  ln -sf /usr/bin/python3.10 /usr/bin/python && \
66
  python3 -m pip install --upgrade pip
 
61
  ca-certificates libglib2.0-0 libgl1 \
62
  && apt-get clean && rm -rf /var/lib/apt/lists/*
63
 
64
+
65
+ RUN apt-get -y install cudnn9-cuda-12
66
+
67
  RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
68
  ln -sf /usr/bin/python3.10 /usr/bin/python && \
69
  python3 -m pip install --upgrade pip