No module named ‘torchvision‘

时间:2025-04-20 10:11:13

系统环境:Win11+Python3.9.7

文章目录

  • 1.介绍
  • 2.问题
  • 3.解决方案
    • (1)pip安装torchvision
    • (2)pip安装指定版本
  • 4.备注

1.介绍

torchvision 是PyTorch中专门用来处理图像的库

2.问题

运行程序时出现No module named ‘torchvision‘错误

3.解决方案

(1)pip安装torchvision

pip install torchvision

(2)pip安装指定版本

pip install torchvision==0.1.8

4.备注

Package Reference

Transforming and augmenting images
Scriptable transforms
Compositions of transforms
Transforms on PIL Image and torch.*Tensor
Transforms on PIL Image only
Transforms on torch.*Tensor only
Conversion Transforms
Generic Transforms
Automatic Augmentation Transforms
Functional Transforms
Models and pre-trained weights
Classification
Semantic Segmentation
Object Detection, Instance Segmentation and Person Keypoint Detection
Video classification
Optical flow
Datasets
Built-in datasets
Base classes for custom datasets
Utils
draw_bounding_boxes
draw_segmentation_masks
draw_keypoints
flow_to_image
make_grid
save_image
Operators
batched_nms
box_area
box_convert
box_iou
clip_boxes_to_image
deform_conv2d
generalized_box_iou
generalized_box_iou_loss
masks_to_boxes
nms
ps_roi_align
ps_roi_pool
remove_small_boxes
roi_align
roi_pool
sigmoid_focal_loss
stochastic_depth
RoIAlign
PSRoIAlign
RoIPool
PSRoIPool
DeformConv2d
MultiScaleRoIAlign
FeaturePyramidNetwork
StochasticDepth
FrozenBatchNorm2d
SqueezeExcitation
Reading/Writing images and videos
Video
Fine-grained video API
Image
Feature extraction for model inspection
API Reference