Skip to content
/ pytorch Public
forked from pytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

License

Notifications You must be signed in to change notification settings

sanpam/pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Environment variables control the disabling of certain explicit SIMD optimizations.

x64 options:
TH_NO_AVX2=1 # disable AVX2 codepaths
TH_NO_AVX=1  # disable AVX codepaths
TH_NO_SSE=1  # disable SSE codepaths

ppc64le options:
TH_NO_VSX=1  # disable VSX codepaths

About

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 30.0%
  • C 22.1%
  • Cuda 21.6%
  • C++ 21.5%
  • CMake 3.3%
  • Fortran 0.8%
  • Other 0.7%