Closed nerdalert closed 1 month ago
@nerdalert I suspect you might have an issue with torch and the cuda version on your machine. In the repo we currently depend on the pypi torch package. According to the install details on https://pytorch.org/, I think that package is compiled for cuda 21.1. The 12.4 version is available using some extra-index-url=https://download.pytorch.org/whl/cu124
.
Setting it up with poetry won't be trivial.
Untested. You could try installing it with
# create a virtual env
python3 -m venv venv
# activate the venv
source venv/bin/active
# install torch with the special index
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
# install this package
pip install -e .
Getting the following on Fedora39 + CUDA. Digging into it, just tracking it here: