大家好,欢迎来到IT知识分享网。
1.安装对应的torch、torchvision
网址:https://pytorch.org/get-started/previous-versions/
搜索对应CUDA版本的安装命令(cu110代表CUDA11.0),在终端中复制命令安装。
安装完之后pip list 查看一下:
import torch print(torch.__version__) print(torch.version.cuda)
2.安装torch-geometric
网址:https://pytorch-geometric.com/whl/
找到对应pytorch版本:
四个库(cluster,scatter,sparse,spline-conv)分别:wget 网页中对应的链接并 pip install 下载好的whl包,即完成安装:
注意自己环境的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
作如下修改:(原文件是第八行,改成了第九行)
终于不报错了。。。(可能在某些情况下还会有问题,暂时够用了,下次再说)
免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://yundeesoft.com/28099.html