栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pytorch torch-sparse安装教程

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

pytorch torch-sparse安装教程

前提:

需要根据自己电脑系统和anaconda环境来下载对应的版本,一定要看仔细了!!!

步骤:

以上述对应版本为例:python3.6,    torch1.6.0,  torchvision0.7.0

1.使用conda安装python3.6虚拟环境
conda create -name pytorch1.6.0 python=3.6

 

查看环境

conda info --envs

 

 激活环境

conda activate pytorch1.6.0 

 

  注:如果不显示环境,conda activate D:Anacondaenvspytorch1.6.0

 2.安装pytorch1.6.0版本
# CUDA 9.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.0
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.0 -c pytorch

# CPU only
conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly -c pytorch

因为我的电脑上没有Cuda,所以用第三条命令。 

查看安装的包

conda list 

 

3.torch_sparse安装

如果先安装torch-geometric,会再次提示安装torch_sparse,故先安装torch_sparse等一系列包。

(pytorch1.6.0) D:Anacondaenvspytorch1.6.0>pip install D:Anacondaenvspytorch1.6.0torch_sparse-0.6.7+cpu-cp36-cp36m-win_amd64.whl

(pytorch1.6.0) D:Anacondaenvspytorch1.6.0>pip install D:Anacondaenvspytorch1.6.0torch_scatter-2.0.4+cpu-cp36-cp36m-win_amd64.whl

(pytorch1.6.0) D:Anacondaenvspytorch1.6.0>pip install D:Anacondaenvspytorch1.6.0torch_cluster-1.5.7+cpu-cp36-cp36m-win_amd64.whl

(pytorch1.6.0) D:Anacondaenvspytorch1.6.0>pip install torch-geometric==1.6.1
  • 需要根据自己电脑系统和anaconda环境来下载对应的版本,一定要看仔细了!!!
  • 下载whl文件的网址:pytorch-geometric官方下载地址
  • pip install torch-geometric 根据下载的文件进行本地安装

注:torch-geometric一定是绝对路径,否则报错以下。

 终于安装完成。

 

 

参考链接:

PyTorch碎片:PyToch和Torchvision对应版本_面壁者-CSDN博客_torchvision和torch的关系

pytorch torch-sparse安装教程_wt-cai的博客-CSDN博客

torch-sparse安装报错,一直安装不上解决方法_附子Jun的博客-CSDN博客

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/280473.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号