Skip to content

Commit

Permalink
Updaate installation steps
Browse files Browse the repository at this point in the history
  • Loading branch information
ksugar committed Aug 31, 2023
1 parent ab8845e commit ae8789d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,14 +23,14 @@ conda activate samapi
If you're using a computer with CUDA-compatible GPU, install `cudatoolkit`.

```bash
conda install -y cudatoolkit=11.7
conda install -c conda-forge -y cudatoolkit=11.7
```

If you're using a computer with CUDA-compatible GPU on Windows, install `torch` with GPU-support with the following command.

```bash
# Windows with CUDA-compatible GPU only
python -m pip install torch --index-url https://download.pytorch.org/whl/cu117
python -m pip install "torch>=1.13.1,<2.0" torchvision --index-url https://download.pytorch.org/whl/cu117
```

Install `samapi` and its dependencies.
Expand Down

0 comments on commit ae8789d

Please sign in to comment.