Santosh-Gupta / SpeedTorch

Library for faster pinned CPU <-> GPU transfer in Pytorch
MIT License
682 stars 39 forks source link

Add cupy to requirements.txt #3

Open mkelly121 opened 5 years ago

Santosh-Gupta commented 5 years ago

Cupy is somehow messy when it comes to checking if it's already installed. So if it's already installed on your system, it might try to reinstall it. For this reason, I recommend installing cupy separately if you don't already have it, rather than having it as a requirement.