search

nextpyp / cet_pick

Cellular content mining and particle localization
https://nextpyp.app/milopyp/
BSD 3-Clause "New" or "Revised" License
5 stars 0 forks source link

Cuda/GPU incompatibilities #1

Open stav-ros opened 4 months ago

stav-ros commented 4 months ago

Not necessarily an issue, but for those with newer GPUs (4090 in my case) requiring newer CUDA, this worked for me:

pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

You also might need to run pip install numpy numba scipy --upgrade (Tutorial) Code runs like butter.

asarnow commented 1 week ago

I also had to set CUDA_VISIBLE_DEVICES, and the above was working.