6.4.1. Environment Dependencies
This section introduces the environment dependencies required for the Horizon Plugin Pytorch. We recommend using the Docker environment provided by the X5 algorithm toolchain. For instructions on how to obtain it, please refer to the Development Machine Environment Setup document.
| Dependency | GPU | CPU |
|---|---|---|
| os | Ubuntu20.04 | Ubuntu20.04 |
| cuda | 11.6 | N/A |
| python | 3.10 | 3.10 |
| torch | 1.13.0+cuda-11.6 | 1.13.0+cpu |
| torchvision | 0.14.0+cuda-11.6 | 0.14.0+cpu |
| Recommended GPUs | titan v/2080ti/v100/3090 | N/A |