cuda-convnet2
Nervana 的 Alex Krizhevsky 的 convnet2 的分支,包含几个扩展,包括:
名为 cudanet 的新 Python 后端,用于集成到 Nervana 的框架中
几个新的内核和函数来支持多路成本、GPU 内存的 Python 接口、对非纹理内核的支持、数组和标量最大/最小比较、局部对比度归一化。
基于 pip 或 cmake 的一行安装
额外的检查和修复。
安装
首先确保您已满足所有必需的依赖包,如 cuda-convnet2页面所述。
# Clone this repository.
git clone git@github.com:NervanaSystems/cuda-convnet2.git
cd cuda-convnet2
mkdir build
cd build
cmake ..
make install #f
2021-06-02 20:02:47
1010KB
Cuda
1