首先,需要在github上下载cudamat库,这里我放到云盘上,如果没办法登录github的话,可以在这里下载?
?链接:https://www.aliyundrive.com/s/pJKdqnujeFE
然后我们需要下载安装cudatool,这个地方是踩坑点,首先必须确保自己的电脑是NVidia的显卡,然后在桌面右击鼠标点击英伟达控制面板,在面板中右下角有系统信息,点击进去后点组件按钮
?
然后点击组件,查看自己的NVCUDA这一栏,比如我的NVCUDA 的驱动为11.1.114版本
?然后去英伟达驱动官网下载驱动组件,并进行安装。Tools | NVIDIA Developer
然后需要安装c++buildtool的安装。这里需要注意坑:最好最好下载2017-2019版本下的,不然会报错。
如果找不到或者下不了c++buildtool的话,可以下载visual studio2017-2019任意版本,在下载工具中选择c++组件(一定要选)。
全部安装完后,在visual studio文件夹下把cl.exe文件的路径放到系统变量下。
在anaconda下Lib\site-packages\cudamat文件夹(应该会生成这个库)
重新下载cudamat库,在库文件夹下开启powershell
nvcc -c -O -o cudamat.obj cudamat.cu nvcc -c -O -o cudamat_kernels.obj cudamat_kernels.cu nvcc --shared -o cudamat.dll cudamat.obj cudamat_kernels.obj -lcublas 生成 cudamat.dll 重命名为libcudamat.dll
nvcc -c -O -o learn.obj learn.cu nvcc -c -O -o learn_kernels.obj learn_kernels.cu nvcc --shared -o libcudalearn.dll learn.obj learn_kernels.obj -lcublas 生成libcudalearn.dll
将cudamat-simplify-compile\cudamat的文件全部复制粘贴至C:\ProgramData\Anaconda3\Lib\site-packages\cudamat目录下(全部替换)。
|