Update README.md
Browse files
README.md
CHANGED
@@ -520,8 +520,8 @@ pip install flash-attn --no-build-isolation
|
|
520 |
```
|
521 |
|
522 |
> [!TIP]
|
523 |
-
> Should you face any difficulties installing the above packages, you can
|
524 |
-
> `pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel
|
525 |
|
526 |
### Audio Input
|
527 |
|
|
|
520 |
```
|
521 |
|
522 |
> [!TIP]
|
523 |
+
> Should you face any difficulties installing the above packages, you can try installing within this Docker container instead:
|
524 |
+
> `pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel`, whose cuda and torch environments have been tested working.
|
525 |
|
526 |
### Audio Input
|
527 |
|