Update README.md

README.md

@@ -521,8 +521,8 @@ pip install flash-attn --no-build-isolation
 ```
 
 > [!TIP]
-> Example: You can install the above requirements in this docker:
-> `pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel`
+> Should you face any difficulties installing the above packages, you can try installing within this Docker container instead:
+> `pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel`, whose cuda and torch environments have been tested working.
 
 ### Audio Input