conda create -n torchmdnet
conda activate torchmdnet
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
conda install pyyaml ipython scikit-learn tqdm
pip install pytorch-lightning
pip install torchmdnet
pip install moleculekit
conda install seaborn pandas jupyter
Check the jupyter notebook in the tutorial folder