Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Responsible AI] Remove torch from responsibleai-vision/text conda file and add in docker file #3499

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
Open
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,9 @@ RUN pip install git+https://github.com/huggingface/evaluate.git
RUN pip install 'Werkzeug>=3.0.3'
RUN pip install 'tqdm>=4.66.3'

# To resolve CVE-2024-5480 vulnerability issue for torch < 2.2.2
RUN pip install 'torch>=2.2.2'

RUN pip freeze

# This is needed for mpi to locate libpython
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,4 +26,3 @@ dependencies:
- datasets
- typing-extensions>=4.8.0
- gevent-websocket
- torch>=2.2.2
Original file line number Diff line number Diff line change
Expand Up @@ -26,5 +26,4 @@ dependencies:
- ipython<8.13
- typing-extensions>=4.8.0
- gevent-websocket
- torch>=2.2.2
- opencv-python-headless==4.8.1.78
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
azure-ai-ml==0.1.0b4
azure-ai-ml==1.21.0
azure.identity==1.10.0
azureml-core==1.43.0
Loading