how to build form source with CUDA

rusty1s / pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

MIT License

1.01k stars 147 forks source link

how to build form source with CUDA #349

Closed asher-bit closed 1 year ago

asher-bit commented 1 year ago

hello, i try to build pytorch-sparse from source code with FORCE_CUDA=1 CUDA_HOME=$CONDA_PREFIX pip install --verbose torch-sparse --no-cache-dir FORCE_ONLY_CUDA=1 CUDA_HOME=/data/jiaoshengjie/anaconda3/envs/spcac/bin python setup.py install however i can only install the cpu_version. I have install cudatoolkit in my conda env but I can't install the cuda version pytorch_sparse. How can i solve this problem

asher-bit commented 1 year ago

I try to use ubuntu20.04 with GLIBC=2.29 to skip install with source code. Thank you for your great work!