Update Torch and TorchVision to more secure versions (#4830)

Author: nargokul

prepare_data/sm-ground_truth_rlhf_llm_finetuning.ipynb

@@ -57,7 +57,7 @@
    "outputs": [],
    "source": [
     "!git clone https://github.com/CarperAI/trlx.git\n",
-    "!pip install torch==2.0.0 --extra-index-url https://download.pytorch.org/whl/cu116 # for cuda\n",
+    "!pip install torch>=2.6.0 --extra-index-url https://download.pytorch.org/whl/cu116 # for cuda\n",
     "%cd trlx\n",
     "!git checkout 355c9741f2e606de796f5c6f9b682f7dd00f97c5\n",
     "!pip install -e .\n",

build_and_train_models/sm-training_compliler_single_node_single_gpu_bert/sm-training_compliler_single_node_single_gpu_bert.ipynb

@@ -99,9 +99,7 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "!pip install \"sagemaker>=2.133.0\" botocore boto3 awscli s3fs typing-extensions \"torch==1.12.0\" --upgrade"
-   ]
+   "source": "!pip install \"sagemaker>=2.133.0\" botocore boto3 awscli s3fs typing-extensions \"torch>=2.6.0\" --upgrade"
   },
   {
    "cell_type": "code",

deploy_and_monitor/sm-batch_transform_with_torchserve/docker/Dockerfile

@@ -26,7 +26,7 @@ RUN cd /tmp \
     && python3 get-pip.py
 
 RUN python -m pip install --no-cache-dir torchserve
-RUN python -m pip install --no-cache-dir torch==1.7.1
+RUN python -m pip install --no-cache-dir torch>=2.6.0
 
 COPY dockerd-entrypoint.sh /usr/local/bin/dockerd-entrypoint.sh
 RUN chmod +x /usr/local/bin/dockerd-entrypoint.sh

deploy_and_monitor/sm-mme_with_torchserve/sm-mme_with_torchserve.ipynb

@@ -241,9 +241,7 @@
    "id": "aa1bdc87",
    "metadata": {},
    "outputs": [],
-   "source": [
-    "!pip install -U torch==2.2.0 diffusers==0.13.0 transformers"
-   ]
+   "source": "!pip install -U torch>=2.6.0 diffusers==0.13.0 transformers"
   },
   {
    "cell_type": "code",

responsible_ai/sm-clarify_text_explainability_text_sentiment_analysis/code/requirements.txt

@@ -1,3 +1,3 @@
 transformers==4.40.0
-torch==2.2.0
+torch>=2.6.0
 pandas

responsible_ai/sm-clarify_text_explainability_text_sentiment_analysis/sm-clarify_text_explainability_text_sentiment_analysis.ipynb

@@ -115,7 +115,7 @@
    },
    "outputs": [],
    "source": [
-    "!pip install \"torch==2.2.0\" --upgrade --quiet\n",
+    "!pip install \"torch>=2.6.0\" --upgrade --quiet\n",
     "!pip install captum --upgrade --quiet"
    ]
   },

archived/albert-base-v2/albert-base-v2.ipynb

@@ -79,9 +79,7 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "!pip install botocore boto3 awscli s3fs typing-extensions pandas numpy \"sagemaker>=2.133.0\" \"torch==1.12.0\" --upgrade"
-   ]
+   "source": "!pip install botocore boto3 awscli s3fs typing-extensions pandas numpy \"sagemaker>=2.133.0\" \"torch>=2.6.0\" --upgrade"
   },
   {
    "cell_type": "code",

archived/deploy_huggingface_model_on_Inf1_instance/inf1_bert_compile_and_deploy.ipynb

@@ -65,7 +65,7 @@
     "!pip install -U transformers==4.15.0\n",
     "!pip install -U sagemaker\n",
     "# In this example, we are using torch 1.9 which is the latest torch version Neo supports for inferentia target\n",
-    "!pip install -U torch==1.9.1\n",
+    "!pip install -U torch>=2.6.0\n",
     "!pip install -U sagemaker"
    ]
   },

archived/nlp_score_dashboard_sec/sec-dashboard/SEC_Section_Extraction_Functions.ipynb

@@ -46,7 +46,7 @@
    "source": [
     "# %%capture\n",
     "!pip install transformers==4.11.3 --ignore-installed ruamel-yaml\n",
-    "!pip install torch==1.9.1 --no-cache-dir\n",
+    "!pip install torch>=2.6.0--no-cache-dir\n",
     "!pip install altair==4.1.0\n",
     "!pip install vega==3.5.0"
    ]

archived/pytorch_mnist_elastic_inference.ipynb

@@ -90,9 +90,7 @@
     "scrolled": true
    },
    "outputs": [],
-   "source": [
-    "pip install torchvision==0.5.0 --no-cache-dir"
-   ]
+   "source": "pip install torch>=2.6.0 --no-cache-dir"
   },
   {
    "cell_type": "code",