pytorch
https://github.com/pytorch/pytorch
Python
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported925 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- BatchNorm1d does not support batchsize>65535 in eval mode with 3 dimension (NxCxL), raise CUDNN_STATUS_NOT_SUPPORTED
- Importing open3d after PyTorch causes free(): invalid pointer
- why my personal compiled libtorch is so slow? 2~3 times slower than caffe
- Different behavior when trace model.
- How to load PyTorch model with LSTM using C++ api
- [jit] requires_grad in JIT constructor/factories
- Massive memory overhead over NumPy
- CUDA optimization: using `__restrict__` whenever possible
- GCP Base Image Wrong CUDA Version
- RNN module weights not compacted
- Docs
- Python not yet supported