cuda
Browse files- Dockerfile +4 -6
Dockerfile
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
FROM
|
2 |
|
3 |
WORKDIR /code
|
4 |
COPY ./requirements.txt /code/requirements.txt
|
@@ -6,8 +6,9 @@ COPY ./requirements.txt /code/requirements.txt
|
|
6 |
#RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
|
7 |
#RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
|
8 |
#https://download.pytorch.org/whl/torch_stable.ht
|
9 |
-
|
10 |
-
|
|
|
11 |
|
12 |
RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113
|
13 |
# --index-url https://download.pytorch.org/whl/cu117
|
@@ -31,9 +32,6 @@ WORKDIR /..
|
|
31 |
# RUN mim install "mmpose>=1.1.0"
|
32 |
RUN pip install -r /code/requirements.txt
|
33 |
|
34 |
-
#https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
|
35 |
-
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
|
36 |
-
|
37 |
# Set up a new user named "user" with user ID 1000
|
38 |
RUN useradd -m -u 1000 user
|
39 |
|
|
|
1 |
+
FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
|
2 |
|
3 |
WORKDIR /code
|
4 |
COPY ./requirements.txt /code/requirements.txt
|
|
|
6 |
#RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
|
7 |
#RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
|
8 |
#https://download.pytorch.org/whl/torch_stable.ht
|
9 |
+
|
10 |
+
#https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
|
11 |
+
RUN apt-get update && apt-get install python3.9 ffmpeg libsm6 libxext6 -y
|
12 |
|
13 |
RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113
|
14 |
# --index-url https://download.pytorch.org/whl/cu117
|
|
|
32 |
# RUN mim install "mmpose>=1.1.0"
|
33 |
RUN pip install -r /code/requirements.txt
|
34 |
|
|
|
|
|
|
|
35 |
# Set up a new user named "user" with user ID 1000
|
36 |
RUN useradd -m -u 1000 user
|
37 |
|