* add cache for huggingface

Signed-off-by: Yang Zhang <yangzhang@nvidia.com>

* change cache location

Signed-off-by: Yang Zhang <yangzhang@nvidia.com>

Co-authored-by: Eric Harper <complex451@gmail.com>
This commit is contained in:
Yang Zhang 2021-10-28 17:04:45 -07:00 committed by GitHub
parent 51253e20be
commit 42b167ee2e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

2
Jenkinsfile vendored
View file

@ -2,7 +2,7 @@ pipeline {
agent {
docker {
image 'gitlab-master.nvidia.com/dl/dgx/pytorch:21.10-py3-devel'
args '--device=/dev/nvidia0 --gpus all --user 0:128 -v /home/TestData:/home/TestData -v $HOME/.cache/torch:/root/.cache/torch --shm-size=8g'
args '--device=/dev/nvidia0 --gpus all --user 0:128 -v /home/TestData:/home/TestData -v $HOME/.cache/torch:/root/.cache/torch -v $HOME/.cache/huggingface/transformers:/root/.cache/huggingface/transformers --shm-size=8g'
}
}
options {