Skip to content

Commit

Permalink
Update pytorch install instructions
Browse files Browse the repository at this point in the history
Pytorch 1.13+ use pytorch-cuda instead of cudatoolkit
  • Loading branch information
DarnellGranberry authored Jan 8, 2025
1 parent 0aee22d commit 1ca87c4
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -75,13 +75,13 @@ To install the precompiled Topaz package and its dependencies, including pytorch
```
conda install topaz fsspec -c tbepler -c pytorch
```
This installs pytorch from the official channel. To install pytorch for specific cuda versions, you will need to add the 'cudatoolkit=X.X' package. E.g. to install pytorch for CUDA 9.0:
This installs pytorch from the official channel. To install pytorch for specific cuda versions, you will need to add the 'pytorch-cuda=X.X' package. E.g. to install for CUDA 11.8:
```
conda install cudatoolkit=9.0 -c pytorch
conda install pytorch-cuda=11.8 -c pytorch -c nvidia
```
or combined into a single command:
```
conda install topaz fsspec cudatoolkit=9.0 -c tbepler -c pytorch
conda install topaz fsspec pytorch-cuda=11.8 -c tbepler -c pytorch -c nvidia
```
See [here](https://pytorch.org/get-started/locally/) for additional pytorch installation instructions.

Expand Down

0 comments on commit 1ca87c4

Please sign in to comment.