vcpkg/ports/libtorch/vcpkg.json
2024-12-17 14:59:52 -05:00

145 lines
2.7 KiB
JSON

{
"name": "libtorch",
"version": "2.1.2",
"port-version": 8,
"description": "Tensors and Dynamic neural networks in Python with strong GPU acceleration",
"homepage": "https://pytorch.org/",
"license": null,
"supports": "(windows & !static) | osx | linux",
"dependencies": [
"blas",
"cpuinfo",
"eigen3",
"fftw3",
"flatbuffers",
{
"name": "flatbuffers",
"host": true
},
"fmt",
"foxi",
"foxi",
"fp16",
"gemmlowp",
"gflags",
"glog",
"lapack",
"lmdb",
"mimalloc",
"onnx",
"onnx-optimizer",
"opencl",
"pocketfft",
"protobuf",
{
"name": "protobuf",
"host": true
},
"pthreadpool",
"sleef",
{
"name": "vcpkg-cmake",
"host": true
},
{
"name": "vcpkg-cmake-config",
"host": true
},
{
"name": "vcpkg-get-python-packages",
"host": true
},
"xnnpack"
],
"default-features": [
{
"name": "cuda",
"platform": "windows & !x86"
},
"opencv",
"zstd"
],
"features": {
"cuda": {
"description": "Build with CUDA GPU backend",
"dependencies": [
"cuda",
"cudnn",
"magma",
"nvidia-cutlass"
]
},
"dist": {
"description": "Use distributed with MPI, Gloo, libuv, TensorPipe",
"dependencies": [
{
"name": "gloo",
"platform": "linux"
},
{
"name": "libtorch",
"default-features": false,
"features": [
"mpi"
],
"platform": "linux"
},
{
"name": "libuv",
"platform": "windows | osx"
},
{
"name": "tensorpipe",
"platform": "linux | osx"
}
]
},
"leveldb": {
"description": "Build with LevelDB",
"dependencies": [
"leveldb",
"snappy"
]
},
"llvm": {
"description": "Build with LLVM",
"dependencies": [
"llvm"
]
},
"mpi": {
"description": "Build with MPI",
"dependencies": [
"mpi"
]
},
"nnpack": {
"description": "Build with NNPack",
"supports": "linux | osx",
"dependencies": [
"nnpack"
]
},
"opencv": {
"description": "Build with OpenCV 3.x",
"dependencies": [
"opencv"
]
},
"vulkan": {
"description": "Build with Vulkan GPU backend",
"dependencies": [
"vulkan",
"vulkan-loader",
"vulkan-memory-allocator"
]
},
"zstd": {
"description": "Build with ZSTD",
"dependencies": [
"zstd"
]
}
}
}