libtorch v2.7.1
- Created
- Updated
Tensors and Dynamic neural networks in Python with strong GPU acceleration
# License
Unspecified
# Supported Platforms
(windows & !static) | osx | linux
- ✅ linux
- ✅ osx
- ✅ windows-dynamic
- ❌
uwp - ❌
windows-static
# Features
Default features: gflags , glog , opencv , zstd
# cuda
Build with CUDA GPU backend
Supports: (windows & x64 & !staticcrt) | (linux & x64)
- ✅ linux
- ✅ windows-x64-dynamic
- ❌
osx - ❌
windows-arm64-dynamic - ❌
windows-x86-dynamic
Dependencies:
- cuda
- cudnn
gloo
Platforms:
linux
Features:cuda- magma
nccl
Platforms:
linux- nvidia-cutlass
tensorpipe
Platforms:
linux
Features:cuda
and 8 transitive dependencies:
- blas (by magma )
- lapack (by magma )
lapack-reference (by lapack , lapack-reference )
Platforms:
!windows | !static | mingw
Features:blas-select,noblas- libnop (by tensorpipe )
- libuv (by tensorpipe )
vcpkg-gfortran (by lapack-reference )
Platforms:
windows & !mingw
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by gloo )
- vcpkg-cmake-config (by gloo )
# dist
Use distributed with MPI, Gloo, libuv, TensorPipe
Supports: linux | windows
- ✅ linux
- ✅ windows-dynamic
- ❌
osx
Dependencies:
gloo
Platforms:
linuxlibtorch
Platforms:
linux
Features:mpilibuv
Platforms:
windows | osxtensorpipe
Platforms:
linux
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by gloo )
- vcpkg-cmake-config (by gloo )
# gflags default
Build with gflags
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by gflags )
- vcpkg-cmake-config (by gflags )
# glog default
Build with glog
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by gflags )
- vcpkg-cmake-config (by gflags )
# llvm
Build with LLVM
Dependencies:
Host Dependencies:
3 transitive dependencies:
- vcpkg-cmake (by llvm )
- vcpkg-cmake-config (by llvm )
- vcpkg-cmake-get-vars (by llvm )
# mpi
Build with MPI
# nnpack
Build with NNPack
Supports: linux | osx
- ✅ linux
- ✅ osx
- ❌
windows-dynamic
Dependencies:
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by cpuinfo )
- vcpkg-cmake-config (by cpuinfo )
# opencv default
Build with OpenCV
Dependencies:
Host Dependencies:
3 transitive dependencies:
- vcpkg-cmake (by opencv4 )
- vcpkg-cmake-config (by opencv4 )
- vcpkg-get-python-packages (by opencv4 )
# vulkan
Build with Vulkan GPU backend
Dependencies:
and one transitive dependency:
- vulkan-headers (by vulkan )
Host Dependencies:
and 5 transitive dependencies:
- glslang (by shaderc )
- spirv-headers (by spirv-tools )
- spirv-tools (by shaderc )
- vcpkg-cmake (by glslang )
- vcpkg-cmake-config (by glslang )
# zstd default
Build with ZSTD
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by zstd )
- vcpkg-cmake-config (by zstd )
# Dependencies
- blas
- cpp-httplib
- cpuinfo
- eigen3
fbgemm
Platforms:
x64- flatbuffers
- fmt
- foxi
- fp16
- gemmlowp
- lapack
- lmdb
- mimalloc
- nlohmann-json
- onnx
- onnx-optimizer
- opencl
- pocketfft
- protobuf
- pthreadpool
- pybind11
- sleef
- xnnpack
and 23 transitive dependencies:
- abseil (by protobuf )
- asmjit (by fbgemm )
- brotli (by cpp-httplib )
- fxdiv (by pthreadpool )
gettext-libintl (by python3 )
Platforms:
!windowslapack-reference (by lapack , lapack-reference )
Platforms:
!windows | !static | mingw
Features:blas-select,noblaslibiconv (by gettext-libintl )
Platforms:
!linux- psimd (by fp16 )
- utf8-range (by protobuf )
vcpkg-gfortran (by lapack-reference )
Platforms:
windows & !mingw- zlib (by python3 )
# Host Dependencies
and 7 transitive dependencies:
- pkgconf (by vcpkg-pkgconfig-get-modules )
vcpkg-cmake-get-vars (by libiconv )
Platforms:
android- vcpkg-get-python (by python3 )
- vcpkg-make (by gettext-libintl )
vcpkg-msbuild (by python3 )
Platforms:
windowsvcpkg-pkgconfig-get-modules (by python3 )
Platforms:
windows & !mingw- vcpkg-tool-meson (by pkgconf )
# Dependents
# Host Dependents
No dependents.
# Contributors
PARK DongHa
jim wang
Lily Wang
MonicaLiu
Ómar Högni Guðmarsson
Alexander Neumann
Jia Yue Hua
Billy O'Neal
miyanyan
JoergAtGithub
Stefano Sinigardi
Mengna-Li
Steve Brain