Skip to content

Latest commit

 

History

14 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tensorplusplus

一个带自动求导的c++的神经网络库,目前仅支持CUDA设备,完全支持数据类型FP64,FP32,BF16,半支持数据类型FP16,编译需要调用

h: cu_plusplus.h

lib: cu_plusplus.lib

dll: cuda toolkit(cublas64_12.dll cublasLt64_12.dll cudart64_12.dll curand64_10.dll)

cudnn(cudnn_adv64_9.dll,cudnn_cnn64_9.dll,cudnn_engines_precompiled64_9.dll,cudnn_engines_runtime_compiled64_9.dll,cudnn_engines_tensor_ir64_9.dll,cudnn_ext64_9.dll,cudnn_graph64_9.dll,cudnn_heuristic64_9.dll,cudnn_ops64_9.dll,cudnn64_9.dll)

底层核函数实现源码cu_plusplus: https://github.com/wz043/cu_plusplus/

使用c++17标准

通过#include "Tensorplusplus.h"开始

About

基于cuda加速带有自动求导的c++深度学习库

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages