Can not use cuda?
XH-B opened this issue · 1 comments
XH-B commented
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_py38h36eccb8_2
how to use cuda? Thanks.
XH-B commented
I try again, i works.