cuda
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
@@ -7,7 +7,8 @@ COPY ./requirements.txt /code/requirements.txt
|
|
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 |
-
RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/
|
|
|
11 |
# --extra-index-url https://download.pytorch.org/whl/cu113
|
12 |
RUN pip install --no-cache-dir --upgrade -U openmim
|
13 |
RUN mim install --no-cache-dir --upgrade mmengine
|
|
|
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 |
+
RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113
|
11 |
+
# --index-url https://download.pytorch.org/whl/cu117
|
12 |
# --extra-index-url https://download.pytorch.org/whl/cu113
|
13 |
RUN pip install --no-cache-dir --upgrade -U openmim
|
14 |
RUN mim install --no-cache-dir --upgrade mmengine
|