![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
Henry Fox
这个作者很懒,什么都没留下…
展开
-
pytorch加载参数设备选择 2021-12-12
https://blog.csdn.net/bc521bc/article/details/85623515由于pytorch的模型保存的时候记录了device信息,所以加载的时候需要手动指定设备。通过:, map_location={'cuda:1':'cuda:0'}的写法转载 2021-12-12 14:15:43 · 164 阅读 · 0 评论 -
jupyter lab安装和使用 2021-12-09
https://www.cnblogs.com/Tsingwaa/p/14681660.html转载 2021-12-09 20:52:10 · 228 阅读 · 0 评论 -
Anaconda环境用户权限 2021-12-01
背景:anaconda是在公用账户上装的,默认设置下anaconda的环境权限不在我的账号下,所以用自己账号的时候会报错。解决:用sudo chown -R 用户名 /home/公用账户/anaconda3/envs/环境名原创 2021-12-01 13:46:53 · 2234 阅读 · 0 评论 -
pytorch中的稀疏矩阵 2021-09-07
pytorch的系数矩阵结构包括:torch_sparse.tensor.SparseTensor: 该结构非pytorch官方给出的数据结构,在torch_geometric等包中使用。可以通过.to_torch_sparse_coo_tensor()转换为下面的COO tensor(coordinate)结构。torch.sparse包中存在的几种数据结构,详情见 https://pytorch.org/docs/stable/sparse.html 。包括torch.sparse_coo_ten原创 2021-09-07 13:22:04 · 1954 阅读 · 2 评论 -
pytorch模型列表的处理 2021-09-06
在pytorch中可能层数不确定,需要用一个列表的形式进行模型构建。如果直接在Module里面写self.xx=[层1,层2]这样就不能将这些参数放到模型可训练参数列表中。需要用self.xx=torch.nn.ModuleList([层1,层2])...原创 2021-09-06 14:06:06 · 412 阅读 · 0 评论 -
GNN包torch_geometric使用 2021-09-02
torch_geometric包是一个进行GNN实现的非常方便的包,官网是 https://pytorch-geometric.readthedocs.io/en/latest/notes/introduction.html数据集格式使用的数据集的格式如下例所示:>>> import torch>>> from torch_geometric.data import Data>>> edge_index = torch.tensor([[0,原创 2021-09-02 16:13:48 · 2048 阅读 · 5 评论 -
pytorch升级conflict
由于在pytorch1.8.1环境下安装了pytorch-geometric,导致升级pytorch的时候出现问题:解决方案:先卸载pytorch-geometric(conda uninstall pytorch-geometric),安装新版pytorch(conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch)后再安装pytorch-geometric(conda install pytorch-geomet原创 2021-08-20 08:52:00 · 282 阅读 · 0 评论