Skip to content
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

tslcharge/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 28.7%
  • C++ 25.0%
  • Cuda 20.9%
  • C 20.6%
  • CMake 3.4%
  • Fortran 0.8%
  • Other 0.6%