Ywung commited on
Commit
54342b2
1 Parent(s): 5f141ab

try fix datasets install

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -3
Dockerfile CHANGED
@@ -2,13 +2,15 @@ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04 as builder
2
  # FROM nvidia/cuda:11.8.0-devel-ubuntu22.04 as builder
3
 
4
  RUN apt-get update && \
5
- apt-get install --no-install-recommends -y git vim build-essential python3-dev python3-venv && \
6
  rm -rf /var/lib/apt/lists/*
7
 
8
  RUN --mount=target=/root/packages.txt,source=packages.txt apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
9
 
10
- # RUN curl https://pyenv.run | bash
11
- # RUN pyenv install 3.10 && pyenv global 3.10 && pyenv rehash && pip install --no-cache-dir --upgrade pip==22.3.1 setuptools wheel && pip install --no-cache-dir datasets "huggingface-hub>=0.12.1" "protobuf<4" "click<8.1" "pydantic~=1.0"
 
 
12
 
13
  RUN git clone https://github.com/oobabooga/GPTQ-for-LLaMa /build
14
 
 
2
  # FROM nvidia/cuda:11.8.0-devel-ubuntu22.04 as builder
3
 
4
  RUN apt-get update && \
5
+ apt-get install --no-install-recommends -y git vim build-essential curl && \
6
  rm -rf /var/lib/apt/lists/*
7
 
8
  RUN --mount=target=/root/packages.txt,source=packages.txt apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
9
 
10
+ RUN curl https://pyenv.run | bash
11
+ RUN pyenv install 3.10 && pyenv global 3.10 && pyenv rehash
12
+ # RUN pip install --no-cache-dir --upgrade pip==22.3.1 setuptools wheel && \
13
+ # pip install --no-cache-dir datasets "huggingface-hub>=0.12.1" "protobuf<4" "click<8.1" "pydantic~=1.0"
14
 
15
  RUN git clone https://github.com/oobabooga/GPTQ-for-LLaMa /build
16