Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
@@ -6,6 +6,8 @@ COPY ./requirements.txt /code/requirements.txt
|
|
6 |
|
7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
8 |
RUN pip install -U bitsandbytes-cuda110
|
|
|
|
|
9 |
|
10 |
COPY . .
|
11 |
|
|
|
6 |
|
7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
8 |
RUN pip install -U bitsandbytes-cuda110
|
9 |
+
RUN pip install -U bitsandbytes-cuda117
|
10 |
+
RUN pip install -U bitsandbytes
|
11 |
|
12 |
COPY . .
|
13 |
|