设置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
具体下载地址，和存放路径，根据运行提示而定即可。