PyTorch安装及试用 基于Anaconda3

2024年5月21日 139次阅读 来源: pytorch

 

设置Torch国内镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

 

安装PyTorch和TorchVision

conda install pytorch torchvision

 

测试pytorch版本

import torch
import torchvision

print(torch.__version__)
print(torchvision.__version__)

encoder = torchvision.models.resnet34(pretrained=True)

print(type(encoder.conv1.weight))
print(encoder.conv1.weight.shape)

其中这句，encoder = torchvision.models.resnet34(pretrained=True)，会自动下载预训练好的resnet34网络，但是速度较慢，且容易断，还没断点续传功能。
用迅雷下载https://download.pytorch.org/models/resnet34-333f7ec4.pth 到 ~/.torch/models/resnet34-333f7ec4.pth
具体下载地址，和存放路径，根据运行提示而定即可。

 

 

 

    原文作者：pytorch
    原文地址: https://www.cnblogs.com/xbit/p/10293381.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞