sharpenb commited on
Commit
2de37bb
1 Parent(s): 10720c4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -41,10 +41,10 @@ These results were obtained on NVIDIA A100-PCIE-40GB with configuration describe
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
47
- pip install pruna-engine[gpu] --extra-index-url https://pypi.nvidia.com --extra-index-url https://pypi.ngc.nvidia.com
48
  ```
49
  3. Download the model files using one of these three options.
50
  - Option 1 - Use command line interface (CLI):
 
41
 
42
  You can run the smashed model with these steps:
43
 
44
+ 0. Check cuda, torch, packaging requirements are installed. For cuda, check with `nvcc --version` and install with `conda install nvidia/label/cuda-12.1.0::cuda`. For packaging and torch, run `pip install packaging torch`.
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
47
+ pip install pruna-engine[gpu] --extra-index-url https://pypi.nvidia.com --extra-index-url https://pypi.ngc.nvidia.com --extra-index-url https://prunaai.pythonanywhere.com/
48
  ```
49
  3. Download the model files using one of these three options.
50
  - Option 1 - Use command line interface (CLI):