Skip to content

Commit 9c9129c

Browse files
authored
Update Dockerfile.1.11-py38-cuda11.3
1 parent 7bddc8b commit 9c9129c

File tree

1 file changed

+20
-21
lines changed

1 file changed

+20
-21
lines changed

python-pytorch/Dockerfile.1.11-py38-cuda11.3

Lines changed: 20 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -10,26 +10,26 @@ ENV TENSORBOARDX_VERSION=2.1
1010
RUN python3 -m pip uninstall -y torch && \
1111
python3 -m pip install -U pip setuptools
1212
RUN python3 -m pip install --no-cache-dir \
13-
http://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl \
14-
keras4torch==1.2.4 \
15-
pytorch-lightning==1.3.4 \
16-
torch-model-archiver==0.3.0 \
17-
torch-tb-profiler==0.4.0 \
18-
http://download.pytorch.org/whl/cu113/torchaudio-0.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl \
19-
torchmetrics==0.7.2 \
20-
torchserve==0.5.3 \
13+
http://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl \
14+
keras4torch==1.2.4 \
15+
pytorch-lightning==1.3.4 \
16+
torch-model-archiver==0.3.0 \
17+
torch-tb-profiler==0.4.0 \
18+
http://download.pytorch.org/whl/cu113/torchaudio-0.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl \
19+
torchmetrics==0.7.2 \
20+
torchserve==0.5.3 \
2121
torchtext==0.12.0 \
2222
http://download.pytorch.org/whl/cu102/torchvision-0.12.0%2Bcu102-cp38-cp38-linux_x86_64.whl \
23-
http://download.pytorch.org/whl/torch_model_archiver-0.3.0-py2.py3-none-any.whl \
24-
jsonargparse \
25-
torchmetrics \
26-
pyDeprecate \
27-
fsspec[http]>=2021.05.0 \
28-
PyYAML>=5.1 \
29-
pretrainedmodels \
30-
cnn-finetune \
31-
keras4torch \
32-
pytorch-lightning
23+
http://download.pytorch.org/whl/torch_model_archiver-0.3.0-py2.py3-none-any.whl \
24+
jsonargparse \
25+
torchmetrics \
26+
pyDeprecate \
27+
fsspec[http]>=2021.05.0 \
28+
PyYAML>=5.1 \
29+
pretrainedmodels \
30+
cnn-finetune \
31+
keras4torch \
32+
pytorch-lightning
3333

3434
RUN python3 -m pip install --extra-index-url \
3535
https://developer.download.nvidia.com/compute/redist --upgrade nvidia-dali-cuda110
@@ -44,9 +44,8 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
4444
RUN python3 -m pip install --no-cache-dir \
4545
mpi4py==3.1.3 \
4646
mlflow==1.24.0 \
47-
nni==2.6.1 \
48-
tensorboard-plugin-wit \
49-
scikit-nni==0.2.1
47+
nni==2.6.1 \
48+
scikit-nni==0.2.1
5049

5150
RUN apt update && \
5251
apt install ncurses-term && \

0 commit comments

Comments
 (0)