Cuda/GPU incompatibilities

nextpyp / cet_pick

Cellular content mining and particle localization

https://nextpyp.app/milopyp/

BSD 3-Clause "New" or "Revised" License

7 stars 0 forks source link

Open stav-ros opened 5 months ago

stav-ros commented 5 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 month ago

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