Update README.md
Browse files
README.md
CHANGED
|
@@ -40,6 +40,7 @@ These results were obtained on NVIDIA A100-PCIE-40GB with configuration describe
|
|
| 40 |
## Setup
|
| 41 |
|
| 42 |
You can run the smashed model with these steps:
|
|
|
|
| 43 |
0. Check that you have cuda installed. You can do this by running `nvcc --version` or `conda install nvidia/label/cuda-12.1.0::cuda`.
|
| 44 |
1. Install the `pruna-engine` available [here](https://pypi.org/project/pruna-engine/) on Pypi. It might take 15 minutes to install.
|
| 45 |
```bash
|
|
|
|
| 40 |
## Setup
|
| 41 |
|
| 42 |
You can run the smashed model with these steps:
|
| 43 |
+
|
| 44 |
0. Check that you have cuda installed. You can do this by running `nvcc --version` or `conda install nvidia/label/cuda-12.1.0::cuda`.
|
| 45 |
1. Install the `pruna-engine` available [here](https://pypi.org/project/pruna-engine/) on Pypi. It might take 15 minutes to install.
|
| 46 |
```bash
|