Topaz train uses only CPU

Particle Picking
9

I haven’t tried working topaz outside CS. I believe the error is more generalized than CS specific (guessing).
Here’s the script I’m using:

#!/usr/bin/env bash
if command -v conda > /dev/null 2>&1; then
    conda deactivate > /dev/null 2>&1 || true  # ignore any errors
    conda deactivate > /dev/null 2>&1 || true  # ignore any errors
fi
unset _CE_CONDA
unset CONDA_DEFAULT_ENV
unset CONDA_EXE
unset CONDA_PREFIX
unset CONDA_PROMPT_MODIFIER
unset CONDA_PYTHON_EXE
unset CONDA_SHLVL
unset PYTHONPATH
unset LD_PRELOAD
unset LD_LIBRARY_PATH

source /usr/local/anaconda3/conda.sh
conda activate topaz
exec topaz $@

Yes, topaz was installed after activating conda environment.
Pytorch was installed as mentioned in the installation guide. I’m not sure how to change pytorch version and if that’s causing the issue.