课程网站：https://www.youtube.com/watch?v=oG0WUq3bRz0
torch文档：https://pytorch.org/cppdocs/
这门课讲的是在pytorch中调用自己的写的cuda核函数
速度真比pytorch写的快，不过不知道为啥。
-
Notifications
You must be signed in to change notification settings - Fork 0
caixiao-0725/pytorch-cuda
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
combine pytorch with cuda.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published