Skip to content

Commit aba8c94

Browse files
committed
add transformer in pytorch kernel
1 parent 75fb692 commit aba8c94

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

python-pytorch/Dockerfile.1.3-py36-cuda10.1

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade \
2525
python3 -m pip install --no-cache-dir tensorboardX==${TENSORBOARDX_VERSION}
2626

2727
RUN python3 -m pip install --no-cache-dir \
28+
transformers \
2829
jupyter-tensorboard==0.2.0 \
2930
tensorwatch==0.8.10 \
3031
jupyterlab-nvdashboard==0.2.0

python-pytorch/Dockerfile.1.4-py36-cuda10.1

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade \
2626
python3 -m pip install --no-cache-dir tensorboardX==${TENSORBOARDX_VERSION}
2727

2828
RUN python3 -m pip install --no-cache-dir \
29+
transformers \
2930
jupyter-tensorboard==0.2.0 \
3031
tensorwatch==0.8.10 \
3132
jupyterlab-nvdashboard==0.2.0

python-pytorch/Dockerfile.1.5-py36-cuda10.1

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade \
2525
python3 -m pip install --no-cache-dir tensorboardX==${TENSORBOARDX_VERSION}
2626

2727
RUN python3 -m pip install --no-cache-dir \
28+
transformers \
2829
jupyter-tensorboard==0.2.0 \
2930
tensorwatch==0.8.10 \
3031
jupyterlab-nvdashboard==0.2.0

0 commit comments

Comments
 (0)