AdaptiveMotorControlLab / CEBRA

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA
https://cebra.ai
Other
875 stars 66 forks source link

Is CEBRA suitable for Windows system? #93

Closed Albertvicient closed 9 months ago

Albertvicient commented 9 months ago

Is there an existing issue for this?

Bug description

Is the CEBRA suitable for Windows System?

Operating System

operating system

CEBRA version

cebra version

Device type

gpu

Steps To Reproduce

No response

Relevant log output

is the cebra suitable for windows system?

Anything else?

No response

Code of Conduct

MMathisLab commented 9 months ago

Hi @Albertvicient we only tested on Linux and Unix systems (Ubuntu and MacOS), as per our docs:

https://cebra.ai/docs/installation.html

image

Although if you have gpu support it could work on windows, but we don't officially test this at this time.

pderdeyn commented 9 months ago

It works for me on my windows machine. The default conda.yml file works, but for me was CPU only. I'd recommend instead installing pytorch manually first using conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia and then installing the rest. If you do install the whole CEBRA environment first, you might have to uninstall pytorch and then run the above command (that's how I set my environment up)