前提:
需要根据自己 电脑系统和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:\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等一系列包。
(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博客
Original: https://blog.csdn.net/weixin_54096215/article/details/120565345
Author: weixin_54096215
Title: pytorch torch-sparse安装教程
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/710658/
转载文章受原作者版权保护。转载请注明原作者出处!