Python 3.10 & PyTorch 2.1 with CUDA. Use conda to install a virtual environment. The following steps have been tested on Miniconda3. conda create -n loc-cbm python=3.10 conda activate loc-cbm conda ...