Skip to content

Commit

Permalink
document how to install cudaXX variants of pytorch
Browse files Browse the repository at this point in the history
  • Loading branch information
stas00 authored Dec 11, 2018
1 parent ec5017a commit 31f6a2e
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -162,6 +162,15 @@ conda install conda
So if your NVIDIA driver is less than 384, then you can only use `cuda80`. Of course, you can upgrade your drivers to more recent ones if your card supports it.
You can find a complete table with all variations [here](https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html).

If you use NVIDIA driver 410+, you most likely need to install the cuda100 pytorch variant, by:
```bash
conda install -c pytorch pytorch cuda100
```
or if you need a lower version (`cuda90` is installed by default), use:
```bash
conda install -c pytorch pytorch cuda80
```
For other options refer to the complete list of [the available pytorch variants](https://pytorch.org/get-started/locally/).

## Updates

Expand Down

0 comments on commit 31f6a2e

Please sign in to comment.