libtorch v2.12.0
- 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 feature: blas
# blas default
BLAS/LAPACK backend (openblas on linux/windows, Accelerate on macOS). On by default; disable with [core,...] when using [mkl] instead.
Dependencies:
and 4 transitive dependencies:
lapack-reference (by lapack , lapack-reference )
Platforms:
!windows | !static | mingw
Features:blas-select,noblasvcpkg-gfortran (by lapack-reference )
Platforms:
windows & !mingw
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by openblas )
- vcpkg-cmake-config (by openblas )
# 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
- cudnn-frontend
gloo
Platforms:
linux
Features:cuda- magma
nccl
Platforms:
linux- nvidia-cutlass
tensorpipe
Platforms:
linux
Features:cuda
and 9 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 )
- nlohmann-json (by cudnn-frontend )
vcpkg-gfortran (by lapack-reference )
Platforms:
windows & !mingw
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by nlohmann-json )
- vcpkg-cmake-config (by nlohmann-json )
# dist
Use distributed training/inference (Gloo, MPI, libuv, TensorPipe)
Supports: (linux & x64) | windows
- ✅ linux
- ✅ windows-dynamic
- ❌
osx
Dependencies:
gloo
Platforms:
linuxlibuv
Platforms:
windows | osxmpi
Platforms:
linuxtensorpipe
Platforms:
linux
Host Dependencies:
6 transitive dependencies:
- pkgconf (by openmpi )
- vcpkg-cmake (by gloo )
- vcpkg-cmake-config (by gloo )
- vcpkg-cmake-get-vars (by vcpkg-tool-meson )
- vcpkg-make (by openmpi )
- vcpkg-tool-meson (by pkgconf )
# gflags
Build with gflags
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by gflags )
- vcpkg-cmake-config (by gflags )
# glog
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 )
# mkl
Intel performance backend: use oneMKL for BLAS/LAPACK and enable oneDNN (MKLDNN) CPU acceleration via ideep. Mutually exclusive with [blas] (the openblas default-feature) — install as libtorch[core,mkl,...] to skip the openblas backend, otherwise both backends get built and only MKL is linked.
Supports: (windows | linux) & x64 & !uwp
- ✅ linux
- ✅ windows-x64-dynamic
- ❌
osx - ❌
windows-arm64-dynamic - ❌
windows-x86-dynamic
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by onednn )
- vcpkg-cmake-config (by onednn )
# 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 )
# Dependencies
- cpp-httplib
- cpuinfo
- eigen3
fbgemm
Platforms:
x64 & !uwp- flatbuffers
- fmt
- fp16
- kineto
kleidiai
Platforms:
arm64mimalloc
Platforms:
windows | (arm64 & !osx)- nlohmann-json
nnpack
Platforms:
linux | osx- onnx
- opencl
- pocketfft
- protobuf
- pthreadpool
- pybind11
- sleef
- xnnpack
and 18 transitive dependencies:
- abseil (by protobuf )
- asmjit (by fbgemm )
- brotli (by cpp-httplib )
- fxdiv (by nnpack )
gettext-libintl (by python3 )
Platforms:
!windowslibiconv (by gettext-libintl )
Platforms:
!linux- psimd (by fp16 )
- utf8-range (by protobuf )
- 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 libiconv )
vcpkg-msbuild (by python3 )
Platforms:
windowsvcpkg-pkgconfig-get-modules (by python3 )
Platforms:
windows & !mingw- vcpkg-tool-meson (by pkgconf )
# Dependents
No 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