Struggling with torch version for install (Ubuntu)

[thephatmaster]

I'm trying the Linux install (because my only Windows machine is a Surface Go) and have hit an issue at step 4: Install Pytorch and pip packages.

I'm on an old i7 machine with 16GB of RAM and an ancient and useless NVIDIA GPU - GT218 (GeForce G210), OS is Ubuntu 20.04 LTS.

When I: pip3 install -r requirements-torch.txt I get:

Defaulting to user installation because normal site-packages is not writeable
Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/cpu, https://download.pytorch.org/whl/cu128
Ignoring torch: markers 'sys_platform == "darwin"' don't match your environment
Ignoring torchvision: markers 'sys_platform == "darwin"' don't match your environment
Ignoring torch: markers 'sys_platform == "win32"' don't match your environment
Ignoring torchvision: markers 'sys_platform == "win32"' don't match your environment
ERROR: Could not find a version that satisfies the requirement torch==2.7.1+cu128 (from versions: 1.4.0, 1.4.0+cpu, [...] 2.4.1+cpu) 

ERROR: No matching distribution found for torch==2.7.1+cu128

I see people strugglign with this and using Anaconda to set up a specific python environment.

It's been a hot minute since I messed with venvs, and anyway it's probably something basic I'm missing.

Does anyone have any ideas?

[nagadomi]

I think there is no 2.7.1+cu128 binary that matches your Python version or CPU.
In any case, the G210 has too low compute capability(1.2 or 1.3), so it is not supported by recent versions of CUDA.

From the error message, it seems that 2.4.1+cpu exists, so it might work on CPU.
Try

pip3 install torch torchvision

[thephatmaster]

Thanks - it went for torch-2.4.1-cp38 and ran out space on my OS drive (facepalm) installing 100s of MB of CUDA stuff.

Is that normal? or should i have tried to force some CPU-only version?

[nagadomi]

That is normal.
If you wish to install the CPU-only version,

pip uninstall torch torchvision
pip install torch==2.4.1+cpu torchvision==0.19.1+cpu --index-url https://download.pytorch.org/whl/cpu

Also, your environment is Python 3.8. Many libraries have recently stopped supporting it.
If it doesn't work due to Python 3.8, you may want to install Python 3.10 or later using pyenv, or upgrade Ubuntu to 24.04.

[thephatmaster]

Lovely thanks, will give that a go next time I fire up that machine and report back / mark answered.

I haven't messed about with pyenv since the pi4 days when some stuff needed Python 3.10

Yeah an upgrade is probably due. Those LTS versions don't last as long as I think!

[thephatmaster]

Just to say I got all this up and running with 3.10 on pyenv.

I'm blown away by the results. I've (slowly CPU-converted) some cellphone camera home movies, and they're almost as good as my standalone stereoscopic camera.

Really great project, thank you.

I would warn anyone else against trying to set this up on Ubuntu 20.04 because there are lots of changes to system-level packages required. I'm nervous to reboot! I'd better get a full 24.04LTS upgrade done soon.