Hg cache (#3080)
* 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:
parent
51253e20be
commit
42b167ee2e
2
Jenkinsfile
vendored
2
Jenkinsfile
vendored
|
@ -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 {
|
||||
|
|
Loading…
Reference in a new issue