Skip to content

Commit d19c716

Browse files
committed
support TL
1 parent 718cdd1 commit d19c716

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python-pytorch/Dockerfile.1.8-py38-cuda11.1

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,8 @@ RUN python3 -m pip uninstall -y torch && \
2121
pyDeprecate \
2222
fsspec[http]>=2021.05.0 \
2323
PyYAML>=5.1 \
24+
pretrainedmodels \
25+
cnn-finetune \
2426
pytorch-lightning && \
2527
python3 -m pip install --no-cache-dir tensorboardX==${TENSORBOARDX_VERSION}
2628

0 commit comments

Comments
 (0)