Skip to content

Commit 1ca87c4

Browse files
Update pytorch install instructions
Pytorch 1.13+ use pytorch-cuda instead of cudatoolkit
1 parent 0aee22d commit 1ca87c4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,13 +75,13 @@ To install the precompiled Topaz package and its dependencies, including pytorch
7575
```
7676
conda install topaz fsspec -c tbepler -c pytorch
7777
```
78-
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:
78+
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:
7979
```
80-
conda install cudatoolkit=9.0 -c pytorch
80+
conda install pytorch-cuda=11.8 -c pytorch -c nvidia
8181
```
8282
or combined into a single command:
8383
```
84-
conda install topaz fsspec cudatoolkit=9.0 -c tbepler -c pytorch
84+
conda install topaz fsspec pytorch-cuda=11.8 -c tbepler -c pytorch -c nvidia
8585
```
8686
See [here](https://pytorch.org/get-started/locally/) for additional pytorch installation instructions.
8787

0 commit comments

Comments
 (0)