🌟

PyTorch + CUDA の Dockerfile

2021/10/02に公開約1,400字

普通にやる場合

下記からPyTorchとCUDAのバージョンを選ぶ

https://hub.docker.com/r/pytorch/pytorch/tags?page=1&ordering=last_updated

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のバージョンは適切なものに置き換えてください

参考にしたもの

Discussion

ログインするとコメントできます