之前写过一篇文章是关于windows下安装thundersvm的。当初在linux下安装thundersvm很容易,因为我的CUDA正好是9.0版本。所以一句命令 pip install thundersvm 即可安装。
但如果CUDA版本不是9.0,安装就比较麻烦。本文记录的是一种可行的方法,但不是通过自己编译,而是用现成的wheel文件来安装。
首先在github(GitHub - Xtra-Computing/thundersvm: ThunderSVM: A Fast SVM Library on GPUs and CPUs)上下载wheel文件。比如点击:
就可以看到:
选择下载划红线(个人推荐)的wheel文件。
然后在pytorch官网下载对应CUDA版本的命令行:
注意一定要选择pytorch 和 cuda 分开安装的那种才可以。这两个装完后,thundersvm就可以正常运行了。建议生成独立的conda环境再安装。