Skip to content

Commit 52c209a

Browse files
Update pytorch installation instructions
1 parent e3bd691 commit 52c209a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

doc/installation.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,11 @@ or you can create a new environment (here called `micro-sam`) via
2727
```
2828
$ mamba create -c conda-forge -n micro-sam micro_sam
2929
```
30+
if you want to use the GPU you need to install PyTorch from the `pytorch` channel instead of `conda-forge`. For example:
31+
```
32+
$ mamba create -c pytorch -c nvidia -c conda-forge micro_sam pytorch pytorch-cuda=12.1
33+
```
34+
You may need to change this command to install the correct CUDA version for your computer, see [https://pytorch.org/](https://pytorch.org/) for details.
3035

3136
You also need to install napari to use the annotation tool:
3237
```

0 commit comments

Comments
 (0)