Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions doc/source/installation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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.
Note TorchDrug runs on Apple silicon CPUs, but doesn't support `mps` devices.