Skip to content

GitLab

Commit 3341dbeb authored by rusty1s's avatar rusty1s
Browse files

[skip ci]

parent 247cdb00
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ 3341dbeb
...@@ -97,7 +97,7 @@ Then run: ...@@ -97,7 +97,7 @@ Then run:
pip install torch-scatter pip install torch-scatter
``` ```
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