upgrading to latest version of transformers for machine learning in sagemaker?

I'm running some experiments which requires transformer version greater than 4.17. to upgrade to latest , can i simply install newer version in my own container ?

Dockerfile

FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-inference:1.10.2-transformers4.17.0-gpu-py38-cu113-ubuntu20.04

RUN pip install 'transformers==4.24.0'

主题

机器学习和人工智能

标签

AWS 深度学习容器 Amazon SageMaker 机器学习和人工智能

语言

English

clouduser

已提问 1 年前545 查看次数

1 回答

最新
投票最多
评论最多

这些答案有用吗？为正确答案投票，以帮助社区从您的知识中受益。

Hi,

The more canonical way to do such update in Python is to use the file 'requirements.txt' (that should probably also be present in your image at top of your python project) and update the project with the version you need and then run pip install -r requirements.txt.

See https://learnpython.com/blog/python-requirements-file/ for more details

Best,

Didier

专家

Didier_Durand

已回答 1 年前

upgrading to latest version of transformers for machine learning in sagemaker?

相关内容