1、从github:https://github.com/cudamat/cudamat下载cudamat代码:
2、将cudamat文件夹copy到site-packages目录下,比如 (我的目录是D:\ProgramData\Anaconda3\envs\mxnet39\Lib\site-packages)
3、打开命令窗口,在site-packages\cudamat目录下,执行以下下面3条命令,注意区分字母o和数字0:
nvcc -c -O -0 cudamat.obj cudamat.cu
nvcc -c -O -0 cudamat_kernels.obj cudamat_kernels.cu
nvcc --shared -o libcudamat.dll cudamat.obj cudamat_kernels.obj -lcublas
执行结果如下:
4、最后site-packages\cudamat目录下生成了三个新文件:
libcudamat.lib、libcudamat.exp、libcudamat.dll