pytorch1.4.0+torch_scatter+torch_sparse+torch_cluster+torch_spline_conv安装

电玩女神 2022-10-27 12:14 331阅读 0赞

环境：  
CUDA10.0.130  
CUDNN7.6.5  
具体参见Ubuntu显卡驱动安装、CUDA+CUDNN安装文章，本文不赘述。

需要配置环境变量：

vim .bashrc
    
    export PATH=/usr/local/cuda-10.0/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
    export CUDA_HOME=/usr/local/cuda-10.0:$CUDA_HOME
    export CPATH="/usr/local/cuda-10.0/include:$CPATH"
    
    source .bashrc

可通过以下命令查看：

echo $PATH
    echo $CPATH

--------------------

以pytorch1.4.0+cu100为例：

pip --default-timeout=100 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

为避免torch-sparse等安装报错，先行下载：  
https://pytorch-geometric.com/whl/torch-1.4.0.html  
具体根据torch版本改变网址即可。  
下载完毕后，安装如下：

cd 下载
    pip install torch_scatter-2.0.3+cu100-cp36-cp36m-linux_x86_64.whl
    pip install torch_sparse-0.6.0+cu100-cp36-cp36m-linux_x86_64.whl
    pip install torch_cluster-1.5.4+cu100-cp36-cp36m-linux_x86_64.whl
    pip install torch_spline_conv-latest+cu100-cp36-cp36m-linux_x86_64.whl
    pip --default-timeout=100 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch-geometric==1.4.1

以上，安装了torch\_scatter,torch\_sparse,torch\_cluster,torch\_spline\_conv,torch\_geometric包及相关依赖。

发表评论取消回复

表情：

评论列表（有 0 条评论，331人围观）

还没有评论，来说两句吧...

相关阅读

相关 Anaconda安装pytorch

1、打开Anaconda prompt执行下面命令： 1.1 新建一个环境 conda create --name python3X python=3.X （我用的3.7）

梦里梦外;/ 2022年12月21日 14:53/ 0 赞/ 138 阅读

相关 Pytorch安装慢

使用pip安装Pytorch时十分缓慢，有时换源也不能快速下载 [https://pytorch.org/get-started/locally/][https_pytor

客官°小女子只卖身不卖艺/ 2022年11月17日 14:54/ 0 赞/ 159 阅读

相关 python2.7安装pytorch_PyTorch安装

这个系列写了好几篇文章，这是相关文章的索引，仅供参考：以下是相关深度学习工具包的安装，包括Tensorflow, PyTorch, Torch等：首先安装libcupt

约定不等于承诺〃/ 2022年10月22日 12:51/ 0 赞/ 294 阅读

相关 Pytorch(一) Pytorch 的安装

安装步骤 1.检查是否有合适的GPU, 若有安装Cuda与CuDNN 2.下载pytorch安装文件，进行pytorch的安装检查电脑是否有合适的GPU 在桌

水深无声/ 2022年09月15日 05:38/ 0 赞/ 293 阅读

相关安装pytorch

[https://blog.csdn.net/u014546828/article/details/80334448][https_blog.csdn.net_u0145468

末蓝、/ 2022年03月25日 03:14/ 0 赞/ 248 阅读

相关 Pytorch安装入门

一、什么是Pytorch Pytorch的官网：https://pytorch.org/ PyTorch是一个较新的深度学习框架。从名字可以看出，其和Torch不同之处

左手的ㄟ右手/ 2022年02月28日 02:50/ 0 赞/ 302 阅读

相关 pytorch安装记录

转自：[http://www.pianshen.com/article/3827257032/][http_www.pianshen.com_article_382725703

逃离我推掉我的手/ 2022年02月19日 06:00/ 0 赞/ 274 阅读

相关 pytorch安装

安装pytorch之前先查看ubuntu下cuda的版本号，再进行安装 conda install pytorch torchvision cudatoolkit=8

古城微笑少年丶/ 2022年01月25日 15:11/ 0 赞/ 302 阅读

相关 window 安装 PyTorch

![20191009191333910.png][][日萌社][Link 1] [人工智能AI：Keras PyTorch MXNet TensorFlow Paddle

太过爱你忘了你带给我的痛/ 2021年12月19日 14:59/ 0 赞/ 367 阅读

相关安装Pytorch

下载pytorch：[https://pytorch.org/get-started/locally/][https_pytorch.org_get-started_local

向右看齐/ 2021年09月30日 08:34/ 0 赞/ 406 阅读