解决wsl正确安装torch_sparse、torch_scatter的问题

发布时间 2023-08-27 18:44:14作者: 逝痕枫舞

快速解决torch_sparse、torch_scatter安装并正确使用的问题

我们如果直接进行pip install后,会因为pip的机制自动下载最新版本的其他依赖,例如torch等cuda版本。
所以我们需要找到对应自己电脑的cuda版本的模块whl,进行离线安装。

找到对应版本

打开https://pytorch-geometric.com/whl/,找到对应版本模块。其中cu代表cuda版本的意思,cp是python版本的意思。

找不到对应文件,文件名为xxx.so.11

wsl中没必要安装cuda驱动,所以我们直接先下载libcusparse.so.11这个文件,并安装到/anaconda3/lib内,并且打开我们使用的bash文件,例如~/.bashrc等,但是只能本机运行,如果想使用vscode或者pycharm我们最好使用/etc/profile编写lib全局路径。

打开profile文件后,在文件最后输入

export LD_LIBRARY_PATH="/home/tk/anaconda3/lib/"

保存后更新Profile,问题解决。