解决wsl正确安装torch_sparse、torch_scatter的问题
快速解决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,问题解决。