pytorch torch-sparse安装教程

前提:

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

步骤:

pytorch torch-sparse安装教程

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

1.使用conda安装python3.6虚拟环境

conda create --name pytorch1.6.0 python=3.6

pytorch torch-sparse安装教程

查看环境

conda info --envs

pytorch torch-sparse安装教程

激活环境

conda activate pytorch1.6.0 

pytorch torch-sparse安装教程

注:如果不显示环境,conda activate D:\Anaconda\envs\pytorch1.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 

pytorch torch-sparse安装教程

3.torch_sparse安装

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

pytorch torch-sparse安装教程
(pytorch1.6.0) D:\Anaconda\envs\pytorch1.6.0>pip install D:\Anaconda\envs\pytorch1.6.0\torch_sparse-0.6.7+cpu-cp36-cp36m-win_amd64.whl

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

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

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

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

pytorch torch-sparse安装教程

终于安装完成。

pytorch torch-sparse安装教程

参考链接:

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

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

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

Original: https://blog.csdn.net/weixin_54096215/article/details/120565345
Author: weixin_54096215
Title: pytorch torch-sparse安装教程

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/710658/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球