Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 

README.md

CuPy Learn

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.

Requirements

conda install pytorch torchvision -c pytorch
  • CuPy
pip install cupy
  • scikit-cuda
pip install scikit-cuda

Usage

cd test/
pytest