jhennies / CebraEM

MIT License
5 stars 2 forks source link

installation for linux fails on pc without cuda #35

Open mzugravu opened 1 year ago

mzugravu commented 1 year ago

while installing on linux using the install.sh on pc without cuda there's an issue. This is printed as error:

Executing installation script Detecting CUDA version Traceback (most recent call last): File "/Bianca/conda/conda_envs/cebra-em-env/bin/install_torch.py", line 7, in exec(compile(f.read(), file, 'exec')) File "/src/CebraEM/cebra-em-core/bin/install_torch.py", line 17, in custom_pytorch() File "/src/CebraEM/cebra-em-core/cebra_em_core/linux/cuda.py", line 79, in custom_pytorch CUDA_NUMS = CUDA_VERSION.strip().split('.') AttributeError: 'NoneType' object has no attribute 'strip'

Quick fix:

moving lines 79 and 80 in cuda.py inside the else statement just below.