install failed, please help about svox2 HOT 5 CLOSED

The problem here, is that torch couldn't identify your gpu's CC (compute capability). You have to tell it by defining an environment variable TORCH_CUDA_ARCH_LIST to your gpu's CC +'+PTX', e.g. in bash:

TORCH_CUDA_ARCH_LIST=5.0+PTX

Then run pip install . again.

If you don't know your gpu's CC, if you have an NVIDIA gpu you can run nvidia-smi to get the model if you don't know it already, then look it up in this table:

If you don't have an NVIDIA gpu you'll have to look up the CC by it's model on the internet.

Unrelated, it's generally considered bad practise to run pip as root.

from svox2.

@dadaphl sorry, I can’t seem to find the exact article, but I think got it from something on mylifeismymessage.net.

I did find a wikipedia article that might contain your gpu. Look for ‘Compute ability’ or ‘CUDA Compute ability’ in the tables (it looks like it only gives it for the GeForce 8xxx and tesla series’)

from svox2.

The problem here, is that torch couldn't identify your gpu's CC (compute capability). You have to tell it by defining an environment variable TORCH_CUDA_ARCH_LIST to your gpu's CC +'+PTX', e.g. in bash:

TORCH_CUDA_ARCH_LIST=5.0+PTX

Then run pip install . again.

If you don't know your gpu's CC, if you have an NVIDIA gpu you can run nvidia-smi to get the model if you don't know it already, then look it up in this table:

If you don't have an NVIDIA gpu you'll have to look up the CC by it's model on the internet.

Unrelated, it's generally considered bad practise to run pip as root.

It worked! Thanks.

from svox2.

If you don't know your gpu's CC, if you have an NVIDIA gpu you can run nvidia-smi to get the model if you don't know it already, then look it up in this table:

@bolshoytoster could you please post a link to this table? My (fairly new) nvidia gpu is not listed in the screenshot. I assume it might have been updated but i can not find this table on the nvidia site. thanks.

from svox2.

I had to use export TORCH_CUDA_ARCH_LIST="6.1 7.5 8.6" for it to work.

from svox2.