Skip to content

GitLab

Unverified Commit d20383f6 authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #79 from CeadeS/master 

Fixed Driver Not Found Error in Docker
parents f2a58110 863b4d7e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
README.md
View file @ d20383f6
...@@ -58,10 +58,10 @@ $ echo $CPATH ...@@ -58,10 +58,10 @@ $ echo $CPATH
>>> /usr/local/cuda/include:... >>> /usr/local/cuda/include:...
``` ```
Then run: Running in a docker container without nvidia driver, pytorch needs to evaluate the compute capabilities and fails. Ensure in this case that the compute capabilities are set in ENV TORCH_CUDA_ARCH_LIST
``` ```
pip install torch-scatter ENV TORCH_CUDA_ARCH_LIST = "6.0 6.1 7.2+PTX 7.5+PTX"
``` ```
If you are running into any installation problems, please create an [issue](https://github.com/rusty1s/pytorch_scatter/issues). If you are running into any installation problems, please create an [issue](https://github.com/rusty1s/pytorch_scatter/issues).
......
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