Can't install polybeast in WSL2

[nesou2]

When trying to install polybeast on WSL2 I encounter an issue with CUDA. In particular, when running: python setup.py install I get an issue with finding CUDA libraries:

CUDA errror

CMake Error at /home/seb/anaconda3/envs/brint/lib/python3.7/site-packages/cmake/data/share/cmake-3.22/Modules/CMakeDetermineCompilerId.cmake:724 (message): Compiling the CUDA compiler identification source file "CMakeCUDACompilerId.cu" failed.

Compiler: /usr/bin/nvcc

Build flags:

Id flags: -v

The output was:

No such file or directory

I am fairly sure I have installed CUDA and the toolkits at least somewhat correctly, as I was able to install moolib which also required the use of the CUDA compiler. I ran pip install moolib within the same virtual environment that I tried running python setup.py install, and it seemed to manage to find the files correctly and install. Do you have any idea what is going wrong? Thanks for the help!