pytorch torch-sparse安装教程

前提:

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

步骤:

pytorch torch-sparse安装教程_第1张图片

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

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

conda create --name pytorch1.6.0 python=3.6

查看环境

conda info --envs

 pytorch torch-sparse安装教程_第2张图片

 激活环境

conda activate pytorch1.6.0 

  注:如果不显示环境,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 

3.torch_sparse安装

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

pytorch torch-sparse安装教程_第3张图片

(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碎片:PyToch和Torchvision对应版本_面壁者-CSDN博客_torchvision和torch的关系

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

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

你可能感兴趣的:(bug,pytorch,深度学习,python)