Installation
VAI pruner only has GPU version. The following environments are supported.
| ID | Operating System | GPU Environment | Name |
|---|---|---|---|
| 0 | Ubuntu 16.04 | Cuda 9.0, cuDnn 7, Driver 384 or above |
vai_p_caffe-cuda9.0-cudnn7-ubuntu16.04 tensorflow_gpu-1.12.3-cp36-cp36m-linux_x86_64.whl vai_p_darknet-cuda9.0-cudnn7-ubuntu16.04 |
| 1 | Ubuntu 18.04 | Cuda 10.0, cuDnn 7, Driver 410 or above |
vai_p_caffe-cuda10.0-cudnn7-ubuntu18.04 tensorflow_gpu-1.12.3-cp36-cp36m-linux_x86_64.whl vai_p_darknet-cuda10.0-cudnn7-ubuntu18.04 |