Update README.md
Browse files
README.md
CHANGED
|
@@ -222,6 +222,7 @@ To work with our elastic models and compilation tools, you'll need to install `e
|
|
| 222 |
pip install thestage
|
| 223 |
pip install 'thestage-elastic-models[nvidia]' --extra-index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple
|
| 224 |
pip install flash-attn==2.7.3 --no-build-isolation
|
|
|
|
| 225 |
pip uninstall apex
|
| 226 |
|
| 227 |
# or for blackwell support
|
|
|
|
| 222 |
pip install thestage
|
| 223 |
pip install 'thestage-elastic-models[nvidia]' --extra-index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple
|
| 224 |
pip install flash-attn==2.7.3 --no-build-isolation
|
| 225 |
+
pip install tensorrt==10.11.0.33 # for 4090
|
| 226 |
pip uninstall apex
|
| 227 |
|
| 228 |
# or for blackwell support
|