RuntimeError: Not compiled with GPU support

Question

pvti opened this issue a year ago · comments

I tried both conda, pip with many combinations of torch (1.11+) and cuda (10.2, 11.3, 11.6) but have not managed to make it work.

ChenSiyi · Answer 1 · Wed May 24 2023 22:45:37 GMT+0800 (China Standard Time)

I tried both conda, pip with many combinations of torch (1.11+) and cuda (10.2, 11.3, 11.6) but have not managed to make it work.

Hi, I met the same problem, have you solved it?

shekshaa · Answer 2 · Thu Jul 27 2023 00:30:26 GMT+0800 (China Standard Time)

I had a similar issue and solved it by adding the path to cuda to the environment variable CUDA_HOME before installation:

# first set the cuda path
export CUDA_HOME="/usr/local/cuda-X.X"

# install
python3 setup.py build develop

Note that X.X specifies the cuda version, e.g. cuda-11.3

_ · Answer 3 · Thu Jul 27 2023 01:17:24 GMT+0800 (China Standard Time)

For future reference, I found this repo: https://github.com/inspiros/tvdcn/tree/master is fantastic for deformable convolution.