removing torch ad dependency

deepanker13 · Jan 5, 2024 · 744fe70 · 744fe70
1 parent 64a99de
commit 744fe70
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 8 deletions.
diff --git a/sdk/python/Dockerfile.conformance b/sdk/python/Dockerfile.conformance
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # Dockerfile for building the source code of conformance tests
-FROM python:3.10
+FROM python:3.10-alpine
 
 WORKDIR /kubeflow/training-operator/sdk/python
 

diff --git a/sdk/python/kubeflow/storage_init_container/requirements.txt b/sdk/python/kubeflow/storage_init_container/requirements.txt
@@ -1,6 +1,3 @@
-torch>=1.13.1
-torchvision>=0.9.1
-torchaudio>=0.8.1
 einops>=0.6.1
 transformers_stream_generator==0.0.4
 boto3==1.33.9

diff --git a/sdk/python/setup.py b/sdk/python/setup.py
@@ -31,9 +31,6 @@
     "retrying>=1.3.3",
     "boto3>=1.33.9",
     "transformers>=4.20.0",
-    "torch>=1.13.1",
-    "torchvision>=0.9.1",
-    "torchaudio>=0.8.1",
     "einops>=0.6.1",
     "transformers_stream_generator>=0.0.4",
     "peft>=0.3.0",
@@ -75,5 +72,4 @@
     install_requires=REQUIRES,
     tests_require=TESTS_REQUIRES,
     extras_require={"test": TESTS_REQUIRES},
-    dependency_links=["https://download.pytorch.org/whl/cu118"],
 )