facebookresearch / torchbeast

A PyTorch Platform for Distributed RL
Apache License 2.0
734 stars 113 forks source link

Can't install polybeast in WSL2 #41

Open nesou2 opened 2 years ago

nesou2 commented 2 years ago

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!