From c5cd8d6dc0a0a2ecc69e349cd7a1d07e2a4972a5 Mon Sep 17 00:00:00 2001 From: Carlos Garcia Jurado Suarez Date: Mon, 11 Dec 2023 09:06:14 -0800 Subject: [PATCH] Back to torch 2.0.1 --- azureml/conda.yml | 2 +- azureml/eval.yml | 2 +- azureml/generate.yml | 2 +- pyproject.toml | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/azureml/conda.yml b/azureml/conda.yml index ce84fc2..f772397 100644 --- a/azureml/conda.yml +++ b/azureml/conda.yml @@ -15,4 +15,4 @@ dependencies: - xformers - scipy # This works, while installing from pytorch and cuda from conda does not - - torch==2.1.0 \ No newline at end of file + - torch==2.0.1 \ No newline at end of file diff --git a/azureml/eval.yml b/azureml/eval.yml index e64cda2..ea6953b 100644 --- a/azureml/eval.yml +++ b/azureml/eval.yml @@ -33,6 +33,6 @@ environment: # azureml://registries/azureml/environments/acpt-pytorch-2.0-cuda11 # image: nvcr.io/nvidia/pytorch:23.10-py3 conda_file: conda.yml display_name: autodoc_prediction -compute: azureml:v100cluster +compute: azureml:t4cluster experiment_name: evaluation description: | \ No newline at end of file diff --git a/azureml/generate.yml b/azureml/generate.yml index 28d3208..d849fcd 100644 --- a/azureml/generate.yml +++ b/azureml/generate.yml @@ -23,6 +23,6 @@ environment: image: mcr.microsoft.com/azureml/curated/acpt-pytorch-2.0-cuda11.7:21 conda_file: conda.yml display_name: autodoc_prediction -compute: azureml:v100cluster +compute: azureml:t4cluster experiment_name: prediction description: | \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index afb5a23..422c8ff 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,7 +19,7 @@ dependencies = [ "typer", "scipy", # This works, while installing from pytorch and cuda from conda does not", - "torch==2.1.0", + "torch==2.0.1", "transformers>=4.35.2", ]