Learn how to use CuPy to write CUDA Kernel Function and invoke it by PyTorch, how to use scikit-cuda to call cublas function and invoke it by PyTorch.
- Anaconda
- PyTorch
conda install pytorch torchvision -c pytorch
- CuPy
pip install cupy
- scikit-cuda
pip install scikit-cuda
cd test/
pytest