diff --git a/doc/source/installation.rst b/doc/source/installation.rst index 63e44c0..f513609 100644 --- a/doc/source/installation.rst +++ b/doc/source/installation.rst @@ -24,7 +24,7 @@ First, let's install PyTorch. To install ``torch-scatter``, we need to check the version of PyTorch and CUDA. -We can get the version of PyTorch by ``python3 -c "import torch; print(torch.__version__")``. +We can get the version of PyTorch by ``python3 -c "import torch; print(torch.__version__)"``. The version of CUDA can be get by ``nvcc -V``. For example, if our PyTorch is ``1.8.0`` and CUDA is ``10.2``, the command should be @@ -101,4 +101,4 @@ for more details. pip install git+https://github.com/rusty1s/pytorch_cluster.git pip install torchdrug -Note TorchDrug runs on Apple silicon CPUs, but doesn't support `mps` devices. \ No newline at end of file +Note TorchDrug runs on Apple silicon CPUs, but doesn't support `mps` devices.