From 1ca87c456bf1eaa99450e2a76efd933a85ae0f9e Mon Sep 17 00:00:00 2001
From: Darnell Granberry <40174000+DarnellGranberry@users.noreply.github.com>
Date: Wed, 8 Jan 2025 16:48:55 -0500
Subject: [PATCH] Update pytorch install instructions

Pytorch 1.13+ use pytorch-cuda instead of cudatoolkit
---
 README.md | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index aaa98ef..b87991a 100644
--- a/README.md
+++ b/README.md
@@ -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.