torch_geometric安装

摘要:
因为需要跑一些别人的GCN的任务,所以就要安装相关的包;首先要注意cuda版本和pytorch中的对应关系,必须严格对应,比如我的机器上,cuda版本是10.1;pytorch的cuda版本也是10.1,之前其实是10.2,又重装了一遍;pytorch版本是1.4;python版本是3.6;然后在这个网页上https://pytorch-geometric.com/whl/torch-1.4.0.

因为需要跑一些别人的GCN的任务,所以就要安装相关的包;

首先要注意cuda版本和pytorch中的对应关系,必须严格对应,比如我的机器上,cuda版本是10.1;

torch_geometric安装第1张

pytorch的cuda版本也是10.1,之前其实是10.2,又重装了一遍;

torch_geometric安装第2张

pytorch版本是1.4; python版本是3.6;然后在这个网页上

https://pytorch-geometric.com/whl/torch-1.4.0.html

CUDA=cu101
pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-cluster==1.4.5pip install torch-geometric

可以先把各个包按照要求都下下来,然后直接pip install XX就很快了;

这里有个问题需要注意一下,安装torch_geometric,其实默认会安装最新版本(目前是1.6.1),但有些方法在1.4.3之后就不支持了。

比如torch_geometric.utils.scatter_这个方法,在1.5及以后的版本就不支持了,所以代码中有用到这个方法的,要不就另写一个函数代替,要不就把版本退到1.4.3

免责声明:文章转载自《torch_geometric安装》仅用于学习参考。如对内容有疑问,请及时联系本站处理。

上篇React:使用手写签名插件:react-canvas-draw 和 react-signature-canvas七牛云配置二级域名下篇

宿迁高防,2C2G15M,22元/月;香港BGP,2C5G5M,25元/月 雨云优惠码:MjYwNzM=

相关文章

Gitea安装笔记

什么是gitea Gitea 是一个开源社区驱动的轻量级代码托管解决方案,后端采用 Go 编写,采用 MIT 许可证. 官网:https://gitea.io/zh-cn/ 为什么要用gitea 开源免费 系统要求低,节约资源 gitea的安装(centos 7源码安装) 相关环境安装 yum install -y wget yum install -...

linux系统centos7.9安装R(编译安装)

查看系统: [root@centos7 ~]# cat /etc/redhat-release CentOS Linux release 7.9.2009 (Core) [root@centos7 ~]# hostnamectl Static hostname: centos7 Icon name: computer-vm...

iOS企业级应用打包与部署

>iOS企业级的打包与部署相关小细节记录,主要是记录下plist文件的格式,便于以后使用的时候方便的找到。<!--more--> iOS企业级打包的相关前提: 1. 一个企业级iOS开发者账号。 2. 申请发布证书xxx_distribute.cer(主证书) 3. 申请企业打包发布证书xxx_distribute.provision(副...

甘特图收集

vue-gantt-elastic:https://github.com/neuronetio/vue-gantt-elastic dhtmlxgantt:https://dhtmlx.com/blog/use-dhtmlxgantt-vue-js-framework-demo/ https://www.cnblogs.com/pengfei-nie/p/...

UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead.

内部实则是bool。原: idxs = torch.Tensor(idxs_0.shape[0]).byte() idxs[:] = 1 idxs[idxs_0 < 0] = 0 idxs[idxs_end >= int(N / args.gps_step)] = 0 delta_p_gt = delta...

PHP-Windows下搭建PHP-MSF环境【原创】

环境:Windows7 64位php-7.0.19php-swoole-1.9.15php-yac-2.0.2php-redis-3.1.2php-mongodb-1.2.10 遇坑: Cygwin: 不能默认使用 Windows 自带的 mingw git,否则报"fatal: Unable to create temporary file: Resul...