https://github.com/tigercosmos/simple-vgg16-cu

Simple VGG16 implemented in CUDA
https://github.com/tigercosmos/simple-vgg16-cu

cublas cuda cudnn vgg16

Last synced: 3 months ago
JSON representation

Simple VGG16 implemented in CUDA

• Host: GitHub
• URL: https://github.com/tigercosmos/simple-vgg16-cu
• Owner: tigercosmos
• Created: 2020-12-01T07:16:36.000Z (almost 5 years ago)
• Default Branch: master
• Last Pushed: 2020-12-01T08:06:08.000Z (almost 5 years ago)
• Last Synced: 2025-04-14T12:56:03.149Z (6 months ago)
• Topics: cublas, cuda, cudnn, vgg16
• Language: Cuda
• Homepage:
• Size: 2.93 KB
• Stars: 7
• Watchers: 2
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Simple VGG16 in CUDA

This is a simple VGG16 net implemented in CUDA.

Simply run `make; ./a.out` to test the code. You may need to setup the environment for CUDA, cudnn, cublas.

The convolution layers use cudnn, and the FC layer uses cublas.