ThomasBlumet
commited on
Commit
·
5b113c0
1
Parent(s):
03071d5
change to cu18
Browse files- Dockerfile +1 -1
- requirements.txt +2 -2
Dockerfile
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
# For more information, please refer to https://aka.ms/vscode-docker-python
|
| 2 |
#FROM python:3.10-slim
|
| 3 |
|
| 4 |
-
FROM nvidia/cuda:11.
|
| 5 |
|
| 6 |
# Install Python and pip
|
| 7 |
RUN apt-get update && apt-get install -y python3-pip python3-venv
|
|
|
|
| 1 |
# For more information, please refer to https://aka.ms/vscode-docker-python
|
| 2 |
#FROM python:3.10-slim
|
| 3 |
|
| 4 |
+
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
|
| 5 |
|
| 6 |
# Install Python and pip
|
| 7 |
RUN apt-get update && apt-get install -y python3-pip python3-venv
|
requirements.txt
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
huggingface_hub
|
| 2 |
-
--extra-index-url https://download.pytorch.org/whl/
|
| 3 |
torch
|
| 4 |
transformers
|
| 5 |
gradio
|
| 6 |
optimum
|
| 7 |
accelerate
|
| 8 |
-
--extra-index-url https://huggingface.github.io/autogptq-index/whl/
|
| 9 |
auto-gptq
|
|
|
|
| 1 |
huggingface_hub
|
| 2 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 3 |
torch
|
| 4 |
transformers
|
| 5 |
gradio
|
| 6 |
optimum
|
| 7 |
accelerate
|
| 8 |
+
--extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
|
| 9 |
auto-gptq
|