Skip to content

GitLab

"docs/source/zh/optimization/xdit.md" did not exist on "c8bb1ff53ee1210e04861abd41518f197cfd8b3c"
Commit c114e78f authored by rusty1s's avatar rusty1s
Browse files

typo

parent 4ce6a976
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 7 deletions
README.md
View file @ c114e78f
...@@ -70,18 +70,12 @@ $ echo $CPATH ...@@ -70,18 +70,12 @@ $ echo $CPATH
>>> /usr/local/cuda/include:... >>> /usr/local/cuda/include:...
``` ```
Then run Then run:
``` ```
pip install torch-scatter pip install torch-scatter
``` ```
or
```
python setup.py install
```
When running in a docker container without nvidia driver, PyTorch needs to evaluate the compute capabilities and may fail. When running in a docker container without nvidia driver, PyTorch needs to evaluate the compute capabilities and may fail.
In this case, ensure that the compute capabilities are set via `TORCH_CUDA_ARCH_LIST`, *e.g.*: In this case, ensure that the compute capabilities are set via `TORCH_CUDA_ARCH_LIST`, *e.g.*:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment