@@ -10,26 +10,26 @@ ENV TENSORBOARDX_VERSION=2.1
10
10
RUN python3 -m pip uninstall -y torch && \
11
11
python3 -m pip install -U pip setuptools
12
12
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 \
21
21
torchtext==0.12.0 \
22
22
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
33
33
34
34
RUN python3 -m pip install --extra-index-url \
35
35
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 && \
44
44
RUN python3 -m pip install --no-cache-dir \
45
45
mpi4py==3.1.3 \
46
46
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
50
49
51
50
RUN apt update && \
52
51
apt install ncurses-term && \
0 commit comments