CUDA11.0+pytorch+torch-geometric安装教程

CUDA11.0+pytorch+torch-geometric安装教程1.安装对应的torch、torchvision网址:https://pytorch.org/get-started/previous-versions/搜索对应CUDA版本的安装命令(cu110代表CUDA11.0),在终端中复制命令安装。安装完之后piplist 查看一下:i

大家好,欢迎来到IT知识分享网。

1.安装对应的torch、torchvision

网址:https://pytorch.org/get-started/previous-versions/

搜索对应CUDA版本的安装命令(cu110代表CUDA11.0),在终端中复制命令安装。

CUDA11.0+pytorch+torch-geometric安装教程

 安装完之后pip list 查看一下:

CUDA11.0+pytorch+torch-geometric安装教程

import torch
print(torch.__version__) 
print(torch.version.cuda) 

 CUDA11.0+pytorch+torch-geometric安装教程

 2.安装torch-geometric

网址:https://pytorch-geometric.com/whl/

找到对应pytorch版本:

CUDA11.0+pytorch+torch-geometric安装教程

 四个库(cluster,scatter,sparse,spline-conv)分别:wget 网页中对应的链接并 pip install 下载好的whl包,即完成安装:

CUDA11.0+pytorch+torch-geometric安装教程

 注意自己环境的python版本以及linux/win就行 

安装完上面四个库后执行 pip install torch-geometric 

以上安装完成。

完成之后 import torch-geometric 发现报错,报错信息:“No module named ‘torch.profiler”

原因是torch1.10以上的版本才有torch.profiler这个库,但是Torch网址CUDA11.0兼容的选项没有torch1.10以上,那怎么办呢?

解决:

找到报错路径里的文件profile.py

作如下修改:(原文件是第八行,改成了第九行)

CUDA11.0+pytorch+torch-geometric安装教程

  终于不报错了。。。(可能在某些情况下还会有问题,暂时够用了,下次再说)

免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://yundeesoft.com/28099.html

(0)

相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注微信