-
Notifications
You must be signed in to change notification settings - Fork 4
/
Dockerfile
101 lines (81 loc) · 3.84 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# nvidia docker image
# Select base image: https://catalog.ngc.nvidia.com/containers
# https://qiita.com/k_ikasumipowder/items/32bf0bc781cbbdfa2edb#%E8%A4%87%E6%95%B0%E3%81%AE%E3%83%90%E3%83%BC%E3%82%B8%E3%83%A7%E3%83%B3%E3%81%AE%E3%82%B3%E3%83%B3%E3%83%86%E3%83%8A%E3%81%AE%E3%83%86%E3%82%B9%E3%83%88
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
# FROM nvcr.io/nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update -y && apt-get install -y sudo python3 python3-pip git
RUN pip3 install --upgrade pip
# # requirements.txtをコンテナにコピー
# COPY requirements.txt /tmp/requirements.txt
# # pip install で必要なパッケージをインストール
# RUN pip3 install -r /tmp/requirements.txt
# Pythonライブラリのインストール
RUN pip3 install numpy pandas matplotlib scikit-learn jupyterlab
RUN pip3 install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
RUN pip3 install wandb einops_exts rotary_embedding_torch av
RUN pip3 install imageio minerl_navigate tensorflow_gan tensorflow-hub==0.15.0
# https://keep-loving-python.hatenablog.com/entry/2022/03/12/144617
RUN pip3 install tensorflow-probability==0.23.0
RUN pip3 install -q -U einops datasets
# for s6
RUN pip3 install mamba-ssm
# # for t2v
# RUN pip install pip install transformers -U
# RUN pip install pip install sentencepiece sacremoses importlib_metadata
# for squid
RUN pip3 install requests==2.24
########################################################################################
# # 基本イメージ
# FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel
# ENV DEBIAN_FRONTEND noninteractive
# RUN apt-get update -y && apt-get -y upgrade && apt-get install -y \
# wget curl git libgl1-mesa-dev
# # Pythonライブラリのインストール
# # RUN pip install numpy pandas matplotlib scikit-learn jupyterlab
# # RUN pip install torch torchvision
# RUN pip install jaxlib wandb einops_exts rotary_embedding_torch av
# RUN pip install imageio minerl_navigate tensorflow_gan tensorflow-hub==0.15.0
# RUN pip install tensorflow-probability==0.23.0
# RUN pip install -q -U einops datasets
# # for s6
# RUN pip install mamba-ssm
# # for t2v
# RUN pip install pip install transformers -U
# RUN pip install pip install sentencepiece sacremoses importlib_metadata
# # for squid
# RUN pip install requests==2.24
# RUN apt-get install -y unrar-free ffmpeg
# RUN apt-get clean && \
# rm -rf /var/lib/apt/lists/*
# WORKDIR /workspace
# # bash シェルをデフォルトコマンドとして設定
# CMD ["bash"]
########################################################################################
# # S6 以前
# # 基本イメージとしてPython 3のオフィシャルイメージを使用
# FROM python:3.10
# ENV DEBIAN_FRONTEND noninteractive
# RUN apt-get update -y && apt-get -y upgrade && apt-get install -y \
# wget curl git libgl1-mesa-dev
# # Pythonライブラリのインストール
# RUN pip install numpy pandas matplotlib scikit-learn jupyterlab
# RUN pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
# RUN pip install wandb einops_exts rotary_embedding_torch av
# RUN pip install imageio minerl_navigate tensorflow_gan tensorflow-hub==0.15.0
# # https://keep-loving-python.hatenablog.com/entry/2022/03/12/144617
# RUN pip install tensorflow-probability==0.23.0
# RUN pip install -q -U einops datasets
# # for s6
# RUN pip install mamba-ssm
# # for t2v
# RUN pip install pip install transformers -U
# RUN pip install pip install sentencepiece sacremoses importlib_metadata
# # for squid
# RUN pip install requests==2.24
# RUN apt-get install -y unrar-free ffmpeg
# RUN apt-get clean && \
# rm -rf /var/lib/apt/lists/*
# WORKDIR /workspace
# # bash シェルをデフォルトコマンドとして設定
# CMD ["bash"]