yeq6x commited on
Commit
f321e5c
1 Parent(s): d3e21b2
Files changed (1) hide show
  1. Dockerfile +11 -8
Dockerfile CHANGED
@@ -1,19 +1,22 @@
1
- # ベースイメージとしてCUDA対応のUbuntuを使用
2
- FROM nvidia/cuda:12.2.2-base-ubuntu20.04
3
 
4
  # 作業ディレクトリを設定
5
  WORKDIR /app
6
 
7
- # 必要なパッケージをインストール
8
  RUN apt-get update && apt-get install -y --no-install-recommends \
9
- python3-pip \
10
- python3-dev \
 
 
 
 
 
 
11
  && apt-get clean \
12
  && rm -rf /var/lib/apt/lists/*
13
 
14
- # Pythonのシンボリックリンクを作成
15
- RUN ln -s /usr/bin/python3 /usr/bin/python
16
-
17
  # 必要なPythonライブラリをインストールするための依存ファイルをコピー
18
  COPY requirements.txt /app/requirements.txt
19
 
 
1
+ # ベースイメージとしてPython 3.9を使用
2
+ FROM python:3.9-slim
3
 
4
  # 作業ディレクトリを設定
5
  WORKDIR /app
6
 
7
+ # NVIDIA Container Toolkitのインストール
8
  RUN apt-get update && apt-get install -y --no-install-recommends \
9
+ curl \
10
+ gnupg \
11
+ && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
12
+ && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
13
+ sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
14
+ tee /etc/apt/sources.list.d/nvidia-container-toolkit.list \
15
+ && apt-get update \
16
+ && apt-get install -y nvidia-container-toolkit \
17
  && apt-get clean \
18
  && rm -rf /var/lib/apt/lists/*
19
 
 
 
 
20
  # 必要なPythonライブラリをインストールするための依存ファイルをコピー
21
  COPY requirements.txt /app/requirements.txt
22