libtorch v2.1.2
- 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: cuda
, opencv
, zstd
# cuda
Build with CUDA GPU backend
Dependencies:
and 8 transitive dependencies:
- blas (by magma )
- lapack (by magma )
lapack-reference (by blas , lapack , lapack-reference )
Platforms:
!osx & !ios & !uwp & !(arm & windows)
- pthreads (by pthread )
vcpkg-gfortran (by lapack-reference )
Platforms:
windows
Host Dependencies:
2 transitive dependencies:
# dist
Use distributed with MPI, Gloo, libuv, TensorPipe
Dependencies:
gloo
Platforms:
linux
libtorch
Platforms:
linux
Features:mpi
libuv
Platforms:
windows | osx
tensorpipe
Platforms:
linux | osx
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by gloo )
- vcpkg-cmake-config (by gloo )
# leveldb
Build with LevelDB
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by leveldb )
- vcpkg-cmake-config (by leveldb )
# 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 3.x
Dependencies:
and 49 transitive dependencies:
- abseil (by protobuf )
at-spi2-atk (by gtk3 )
Platforms:
linux
- at-spi2-core (by at-spi2-atk )
- atk (by at-spi2-atk )
- brotli (by freetype )
- bzip2 (by freetype )
dbus (by at-spi2-core )
Platforms:
linux
- dirent (by glib )
egl-registry (by libepoxy )
Platforms:
!osx & !windows
- expat (by dbus )
- flatbuffers (by opencv4 )
- fontconfig (by cairo )
freetype (by cairo , fontconfig , harfbuzz )
Features:
brotli
,bzip2
,png
,zlib
- fribidi (by pango )
- gdk-pixbuf (by gtk3 )
- getopt (by fontconfig )
getopt-win32 (by getopt )
Platforms:
windows & !mingw
- gettext (by atk )
- gettext-libintl (by glib )
- glib (by at-spi2-core )
- libcap (by libsystemd )
- libepoxy (by gtk3 )
- libffi (by glib )
libiconv (by gettext-libintl )
Platforms:
!linux
- libjpeg-turbo (by opencv4 )
- liblzma (by libsystemd )
- libmount (by libsystemd )
- libpng (by opencv4 )
libsystemd (by dbus )
Platforms:
linux
libuuid (by fontconfig )
Platforms:
!windows & !osx & !mingw
- libxcrypt (by libsystemd )
- libxml2 (by at-spi2-atk )
- lz4 (by libsystemd )
- pango (by gtk3 )
- pixman (by cairo )
- protobuf (by opencv4 )
- pthread (by cairo )
- pthreads (by pthread )
- quirc (by opencv4 )
- utf8-range (by protobuf )
- zlib (by opencv4 )
- zstd (by libsystemd )
Host Dependencies:
6 transitive dependencies:
- gperf (by libsystemd )
- vcpkg-cmake (by opencv4 )
- vcpkg-cmake-config (by opencv4 )
vcpkg-cmake-get-vars (by libpng )
Platforms:
arm & android
- vcpkg-get-python-packages (by opencv4 )
- vcpkg-tool-meson (by libsystemd )
# vulkan
Build with Vulkan GPU backend
Dependencies:
and one transitive dependency:
- vulkan-headers (by vulkan )
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by vulkan )
- vcpkg-cmake-config (by vulkan-headers )
# zstd default
Build with ZSTD
Host Dependencies:
2 transitive dependencies:
- vcpkg-cmake (by zstd )
- vcpkg-cmake-config (by zstd )
# Dependencies
- blas
- cpuinfo
- eigen3
- fftw3
- flatbuffers
- fmt
- foxi
- foxi
- fp16
- gemmlowp
- gflags
- glog
- lapack
- lmdb
- mimalloc
- onnx
- onnx-optimizer
- opencl
- pocketfft
- protobuf
- pthreadpool
- sleef
- xnnpack
and 10 transitive dependencies:
- abseil (by protobuf )
- fxdiv (by pthreadpool )
lapack-reference (by blas , lapack , lapack-reference )
Platforms:
!osx & !ios & !uwp & !(arm & windows)
- psimd (by fp16 )
- pthreads (by pthread )
- utf8-range (by protobuf )
vcpkg-gfortran (by lapack-reference )
Platforms:
windows
# Host Dependencies
and 18 transitive dependencies:
- gettext-libintl (by gettext )
libiconv (by gettext-libintl )
Platforms:
!linux
- pkgconf (by vcpkg-pkgconfig-get-modules )
- vcpkg-cmake-get-vars (by vcpkg-msbuild )
- vcpkg-get-python (by python3 )
vcpkg-msbuild (by python3 )
Platforms:
windows
- vcpkg-pkgconfig-get-modules (by vcpkg-msbuild )
- vcpkg-tool-meson (by pkgconf )
- zlib (by python3 )
# 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