Specify a cuda build of pytorch, install pytorch via conda rather tha…

…n pip, remove commented out pin of libwevp

conda/environments/all_cuda-125_arch-x86_64.yaml

@@ -98,6 +98,7 @@ dependencies:
 - python-docx==1.1.0
 - python-graphviz
 - python=3.10
+- pytorch=2.4.0=cuda120*
 - rapidjson=1.1.0
 - rapids-dask-dependency=24.10
 - rdma-core>=48
@@ -125,7 +126,6 @@ dependencies:
 - websockets
 - yapf=0.40.1
 - pip:
-  - --extra-index-url https://download.pytorch.org/whl/cu124
   - --find-links https://data.dgl.ai/wheels-test/repo.html
   - --find-links https://data.dgl.ai/wheels/cu121/repo.html
   - databricks-cli < 0.100
@@ -140,5 +140,4 @@ dependencies:
   - nemollm==0.3.5
   - pymilvus==2.3.6
   - pytest-kafka==0.6.0
-  - torch==2.4.0+cu124
 name: all_cuda-125_arch-x86_64

conda/environments/dev_cuda-125_arch-x86_64.yaml

@@ -82,6 +82,7 @@ dependencies:
 - python-docx==1.1.0
 - python-graphviz
 - python=3.10
+- pytorch=2.4.0=cuda120*
 - rapidjson=1.1.0
 - rapids-dask-dependency=24.10
 - rdma-core>=48
@@ -105,11 +106,9 @@ dependencies:
 - websockets
 - yapf=0.40.1
 - pip:
-  - --extra-index-url https://download.pytorch.org/whl/cu124
   - databricks-cli < 0.100
   - databricks-connect
   - milvus==2.3.5
   - pymilvus==2.3.6
   - pytest-kafka==0.6.0
-  - torch==2.4.0+cu124
 name: dev_cuda-125_arch-x86_64

conda/environments/examples_cuda-125_arch-x86_64.yaml

@@ -48,6 +48,7 @@ dependencies:
 - python-docx==1.1.0
 - python-graphviz
 - python=3.10
+- pytorch=2.4.0=cuda120*
 - rapids-dask-dependency=24.10
 - requests
 - requests-cache=1.1
@@ -63,7 +64,6 @@ dependencies:
 - watchdog=3.0
 - websockets
 - pip:
-  - --extra-index-url https://download.pytorch.org/whl/cu124
   - --find-links https://data.dgl.ai/wheels-test/repo.html
   - --find-links https://data.dgl.ai/wheels/cu121/repo.html
   - databricks-cli < 0.100
@@ -77,5 +77,4 @@ dependencies:
   - milvus==2.3.5
   - nemollm==0.3.5
   - pymilvus==2.3.6
-  - torch==2.4.0+cu124
 name: examples_cuda-125_arch-x86_64

conda/environments/runtime_cuda-125_arch-x86_64.yaml

@@ -36,6 +36,7 @@ dependencies:
 - python-confluent-kafka>=1.9.2,<1.10.0a0
 - python-graphviz
 - python=3.10
+- pytorch=2.4.0=cuda120*
 - rapids-dask-dependency=24.10
 - requests
 - requests-cache=1.1
@@ -47,10 +48,8 @@ dependencies:
 - watchdog=3.0
 - websockets
 - pip:
-  - --extra-index-url https://download.pytorch.org/whl/cu124
   - databricks-cli < 0.100
   - databricks-connect
   - milvus==2.3.5
   - pymilvus==2.3.6
-  - torch==2.4.0+cu124
 name: runtime_cuda-125_arch-x86_64

dependencies.yaml

@@ -339,14 +339,14 @@ dependencies:
           - feedparser=6.0
           - grpcio
           - grpcio-status
-          # - libwebp=1.3.2 # Required for CVE mitigation: https://nvd.nist.gov/vuln/detail/CVE-2023-4863 ##
           - mlflow #>=2.10.0,<3
           - mrc=24.10
           - networkx=2.8.8
           - numpydoc=1.5
           - pydantic
           - python-confluent-kafka>=1.9.2,<1.10.0a0
           - python-graphviz
+          - pytorch=2.4.0=cuda120*
           - pluggy=1.3
           - rapids-dask-dependency=24.10 # provides dask and distributed
           - requests
@@ -360,12 +360,10 @@ dependencies:
           - websockets
           - pip
           - pip:
-            - --extra-index-url https://download.pytorch.org/whl/cu124
             - databricks-cli < 0.100
             - databricks-connect
             - milvus==2.3.5 # update to match pymilvus when available
             - pymilvus==2.3.6
-            - torch==2.4.0+cu124
 
   test_python_morpheus:
     common: