search

TuragaLab / DECODE

This is the official implementation of our publication "Deep learning enables fast and dense single-molecule localization with high accuracy" (Nature Methods)
GNU General Public License v3.0
95 stars 26 forks source link

Can not use cuda? #202

Closed XH-B closed 1 year ago

XH-B commented 1 year ago

I follow the documention to intall the pytorch. conda create -n decode_env -c turagalab -c pytorch -c conda-forge decode=0.10.0 cudatoolkit=11.0 jupyterlab ipykernel

but i just intall a CPU version of pytorch _pytorch conda-forge/linux-64::pytorch-1.7.1-cpu_py38h36eccb82

how to use cuda? Thanks.

XH-B commented 1 year ago

I try again, i works.