有关kaggle中更新cuda与pytorch的版本,并安装pyg

Posted 2023-03-02 silent_invisible

1.更新cuda

1.我更新成torch==1.11.0+cu113,在在cell中输入

!pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

标注:environment不要用最新的,需要其他版本可在官网上寻找连接

2.需要其他版本可在官网上寻找连接[https://pytorch.org/get-started/previous-versions/

3.

##查看torch版本
import torch
print(torch.__version__)#torch的版本
print(torch.version.cuda)#cuda的版本
print(torch.cuda.is_available())#cuda与torch是否匹配

2.安装pyg

1.选择cuda和torch相对应的版本的pyg

try:
    import torch_geometric
except:
    import torch
    """这里改成自己需要的版本"""
    CUDA = "cu113"
    TORCH = "1.11.0"
    
    ! pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html --target=/kaggle/working/pygcluster
    ! pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html --target=/kaggle/working/pygscatter
    ! pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html --target=/kaggle/working/pygsparse
    ! pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html --target=/kaggle/working/pygspline
    ! pip install torch-geometric

但这个方法很耗时间(我安装了一个多小时),我也没有找到其他方法来安装
.