mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 08:44:59 +00:00
cleanup torch cpu magic
This commit is contained in:
parent
cd2aca1d74
commit
84f2b0cbfd
1 changed files with 0 additions and 45 deletions
45
setup.py
45
setup.py
|
@ -1,6 +1,4 @@
|
||||||
import re
|
import re
|
||||||
import subprocess
|
|
||||||
import sys
|
|
||||||
|
|
||||||
from setuptools import find_packages, setup
|
from setuptools import find_packages, setup
|
||||||
|
|
||||||
|
@ -14,49 +12,6 @@ def get_requirements():
|
||||||
with open("requirements.txt") as f:
|
with open("requirements.txt") as f:
|
||||||
requirements = f.read().splitlines()
|
requirements = f.read().splitlines()
|
||||||
|
|
||||||
requirements = [line for line in requirements if not line.startswith("---extra-index-url")]
|
|
||||||
|
|
||||||
torch = next((req for req in requirements if req.startswith("torch==")), None)
|
|
||||||
if not torch:
|
|
||||||
return requirements
|
|
||||||
|
|
||||||
pytorch_url = None
|
|
||||||
|
|
||||||
cmd = [
|
|
||||||
sys.executable,
|
|
||||||
"-m",
|
|
||||||
"pip",
|
|
||||||
"install",
|
|
||||||
torch,
|
|
||||||
"--no-deps",
|
|
||||||
"--dry-run",
|
|
||||||
"--no-cache-dir", # Find the true URL even if another env has cached it
|
|
||||||
"--index-url",
|
|
||||||
"https://download.pytorch.org/whl/cpu",
|
|
||||||
]
|
|
||||||
|
|
||||||
# print(' '.join(cmd))
|
|
||||||
|
|
||||||
try:
|
|
||||||
process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
|
|
||||||
for line in process.stdout:
|
|
||||||
url_match = re.search(r"Downloading (https://download\.pytorch\.org/[^\s]+\.whl)", line)
|
|
||||||
if url_match:
|
|
||||||
pytorch_url = url_match.group(1)
|
|
||||||
process.terminate() # Terminate the subprocess
|
|
||||||
break
|
|
||||||
|
|
||||||
process.wait() # Wait for the process to finish
|
|
||||||
except subprocess.CalledProcessError as e:
|
|
||||||
print(f"Error running pip download: {e}")
|
|
||||||
|
|
||||||
# print(pytorch_url)
|
|
||||||
# sys.exit()
|
|
||||||
|
|
||||||
if pytorch_url:
|
|
||||||
requirements.remove(torch)
|
|
||||||
requirements = [f"torch @ {pytorch_url}"] + requirements
|
|
||||||
|
|
||||||
return requirements
|
return requirements
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue