torchvision¶
Datasets, Transforms and Models specific to Computer Vision
homepage: https://github.com/pytorch/vision
Available installations¶
| torchvision version | Supported CPU targets | Supported GPU targets | EESSI version | Module |
|---|---|---|---|---|
| 0.16.0 | generic: aarch64, x86_64Arm: a64fx, neoverse_n1, neoverse_v1, nvidia/graceAMD: zen2, zen3, zen4Intel: haswell, skylake_avx512, sapphirerapids, icelake, cascadelake |
(none) | 2023.06 | torchvision/0.16.0-foss-2023a |