🌟
PyTorch + CUDA の Dockerfile
普通にやる場合
下記からPyTorchとCUDAのバージョンを選ぶ
conda使つかいたくない場合
FROM nvidia/cuda:11.0.3-runtime-ubuntu18.04
WORKDIR /app
RUN apt-get update
RUN apt-get install -y software-properties-common tzdata
ENV TZ=Asia/Tokyo
RUN add-apt-repository ppa:deadsnakes/ppa
RUN apt-get -y install python3.9 python3.9-distutils python3-pip
RUN python3.9 -m pip install -U pip wheel setuptools
RUN python3.9 -m pip install torch==1.7.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html
cuda, os, pytorchのバージョンは適切なものに置き換えてください
参考にしたもの
- 基本方針
https://codechacha.com/ja/ubuntu-install-python39/ - pip を入れる
https://stackoverflow.com/questions/68185770/installing-python3-9-and-pip-in-docker-image-and-not-python-3-8 - タイムゾーン選べって言われてこける
https://sleepless-se.net/2018/07/31/docker-build-tzdata-ubuntu/ - python3.9 -m pip で importエラー
https://stackoverflow.com/questions/63823964/importerror-cannot-import-name-sysconfig-from-distutils-usr-lib-python3-8
ImportError: cannot import name 'sysconfig' from 'distutils' (/usr/lib/python3.9/distutils/__init__.py) - Dockerビルド時にexit code: 137で落ちる
メモリを6GBにした
https://stackoverflow.com/questions/57291806/docker-build-failed-after-pip-installed-requirements-with-exit-code-137
/bin/sh -c pip3 install torch==1.7.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html: exit code: 137
Discussion