Workaround to run plant-seg in colaboratory (with CUDA12.2 as of Jan. 2024)

[totti0223]

F.Y.I.

!pip install -q condacolab
import condacolab
condacolab.install()

!conda install -q -c conda-forge mamba
# https://github.com/conda-incubator/condacolab/issues/60
with open('/usr/local/conda-meta/pinned', 'w') as f:
    f.write("python 3.10.*\npython_abi 3.10.* *cp310*\ncuda-toolkit 12.2.*")
!mamba install -c pytorch -c nvidia -c conda-forge pytorch pytorch-cuda=12.1 pyqt plant-seg

!plantseg --config config.yaml