A Library extending PyTorch for Personal Needs backed by C++/CUDA APIs
Project description
🔥 MyTorch 🔥
🐣 A Library extending PyTorch for Personal Needs backed by C++/CUDA APIs
| 🚧 WIP Forever 🚧 |
|---|
Installation 👨💻
I have not included any dependencies in the setup.py nor a requirements.txt as I leave the hassle of setting up GPU support for torch on your own. It should work on torch>=1.4 and CUDA>=10.0 but I frankly have no clue. I use torch==1.7.1 and CUDA Version of 11.2
To install it, just do
pip install git+https://github.com/Syzygianinfern0/MyTorch.git
Its also available on PyPi, but I wouldn't be very keen on maintaining it.
pip install python-mytorch==0.1
Documentation 📑
mytorch.ops
mytorch.ops.im2col and mytorch.ops.col2im
- Rearrange image blocks into columns.
- The representation is used to perform GEMM-based convolution.
- Output is 5D (or 6D in case of minibatch) tensor.
- Minibatch implementation is inefficient, and could be done in a single CUDA kernel.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file python-mytorch-0.1.2.tar.gz.
File metadata
- Download URL: python-mytorch-0.1.2.tar.gz
- Upload date:
- Size: 4.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.3.0 pkginfo/1.7.0 requests/2.25.1 setuptools/53.0.0 requests-toolbelt/0.9.1 tqdm/4.56.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
f2ae51ee7e3c65851584497bf331820f1bc4eb290a00ba3922c9f988554706b9
|
|
| MD5 |
a559909be65d4dce2cf2f7f05b96015e
|
|
| BLAKE2b-256 |
928e3a90a67af0edc21f26f0a84efe065039c1eb604fb1a16cece684fe25609a
|
File details
Details for the file python_mytorch-0.1.2-py3-none-any.whl.
File metadata
- Download URL: python_mytorch-0.1.2-py3-none-any.whl
- Upload date:
- Size: 4.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.3.0 pkginfo/1.7.0 requests/2.25.1 setuptools/53.0.0 requests-toolbelt/0.9.1 tqdm/4.56.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
77a49c8d923beac34232360c2b05e28bbe86bb1c4d10022fa4c31e5a3b2ea20b
|
|
| MD5 |
2f09999e5178aa87c9c0f89fe8511bed
|
|
| BLAKE2b-256 |
7eaa99d8409110b7bf0d8143c63d8d6e59c72d3e508c6f3bf5c52271f36215d6
|
