diff --git a/tutorials/openmm-torch-nnpops.ipynb b/tutorials/openmm-torch-nnpops.ipynb index 6e2ca0d5..b682d5b6 100644 --- a/tutorials/openmm-torch-nnpops.ipynb +++ b/tutorials/openmm-torch-nnpops.ipynb @@ -84,7 +84,8 @@ "source": [ "!conda install -q -c conda-forge \\\n", " openmm-torch nnpops torchani openmmtools \\\n", - " &> /dev/null # Comment this line to see a log" + " pytorch=1.11 cudatoolkit=11.2 \\\n", + " &> /dev/null # Comment this line to see a log\"" ] }, {