首页 > 其他分享 >解决wsl正确安装torch_sparse、torch_scatter的问题

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

时间:2023-08-27 18:45:03浏览次数:46  
标签:sparse lib torch wsl cuda 版本 安装

快速解决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,问题解决。

标签:sparse,lib,torch,wsl,cuda,版本,安装
From: https://www.cnblogs.com/adroitwolf/p/17660660.html

相关文章

  • Hello Cuda(三)——VSCODE&LibTorch
    CMAKEListscmake_minimum_required(VERSION3.0FATAL_ERROR)project(example)set(CMAKE_PREFIX_PATH/home/xuliangyu/libtorch)set(Torch_DIR/home/shiyanshi/libtorch/share/cmake/)#修改这里为自己的libtorchcmake路径,也可以用pytorch安装时编译的pkg里的cmakefind......
  • Sparse R-CNN
    DETR是一个简洁的pipeline,但有一部分操作还是Dense的DETR提出了一套不同于DensePrediction的pipeline,将检测视为一个SetPrediction问题,成功去掉了AnchorGeneration和NMS但是在Decoder中,ObjectQuery和FeatureMap上每一个点要算一次CrossAttention,这个计算过程同样是Dense的。......
  • ModuleNotFoundError: No module named 'torch._six'
    ModuleNotFoundError:Nomodulenamed'torch._six' ---------------------------------由于torch2.0版本没有这个模块,因此,只需要降低torch版本,就可以解决。1)新建一个虚拟环境2)安装低版本torch REFhttps://blog.csdn.net/wzwddkb/article/details/130400156-------......
  • win10 CUDA11.1安装torch1.9 / reformer_pytorch
    环境NVIDIA-SMI457.52DriverVersion:457.52CUDAVersion:11.1安装torch-gpucondacreate-ntorch1.9python=3.8pipinstalltorch==1.9.1+cu111torchvision==0.10.1+cu111torchaudio==0.9.1-fhttps://download.pytorch.org/whl/torch_stable.htmlc......
  • PyTorch 提高生产力的技巧
    推荐:使用NSDT场景编辑器助你快速搭建3D应用场景介绍您是否曾经花费数小时调试机器学习模型,但似乎找不到准确性没有提高的原因?你有没有觉得一切都应该完美地工作,但由于某种神秘的原因,你没有得到模范的结果?好吧,没有了。作为初学者探索PyTorch可能会令人生畏。在本文中,您将探索......
  • wsl 编译 android AOSP 13 源码
    安装WSLWindows11版本如下:首先在microsoftstore里面安装ubuntu版本,我选择的是ubuntu22.04:安装完成后,打开"控制面板"-->"程序和功能"-->"启动或关闭Windows功能",打开下面的选项:安装成功后迁移WSL到一个可用空间至少300G的磁盘wsl默认安装在C盘,因......
  • 讲解pytorch的tensor没有移除某个元素的操作
    在PyTorch中,要从一个Tensor中移除一个元素,您需要使用索引操作来选择保留的元素,然后重新创建一个新的Tensor。由于PyTorch的Tensor是不可变的,所以无法直接在原Tensor上移除元素。下面是一个示例,展示了如何从一个PyTorchTensor中移除指定位置的元素:importtorch#......
  • Lnton羚通视频算法算力云平台【PyTorch】教程:学习基础知识如何保存和加载模型
    保存和加载模型是指将训练好的神经网络模型保存到文件中,以便在需要时重新加载该模型进行预测、推断或继续训练。保存模型的过程是将模型的参数和其他相关信息(如优化器状态等)保存到文件中。通过保存模型,我们可以在不重新训练的情况下保留模型的状态,方便后续使用。加载模型的过程是从......
  • Lnton羚通视频算法算力云平台【PyTorch】教程:学习Datasets-DataLoader基础知识
    Dataset&DataLoaderPyTorch提供了两个数据处理的基本方法:torch.utils.data.DataLoadertorch.utils.data.Dataset允许使用预加载的数据集以及自己的数据。Dataset存储样本及其对应的标签,DataLoader在Dataset基础上封装了一个可迭代的对象,以方便访问样本。PyTorch提供了......
  • torch基础操作汇总(常更新)
    已知一个torchtensor[2,4,5,6],要取其中对应的索引位置(2,3),怎么操作你可以使用PyTorch的索引操作来从给定的张量中提取特定索引位置的元素。在你提供的示例中,张量是 [2,4,5,6],你希望提取的索引位置是 (2,3)。以下是如何进行操作:importtorchtensor=torch.tensor(......