pytorch/examples

Segmentation fault (core dumped) at `model(images)` for examples/imagenet/main.py

MaoZiming opened this issue · 1 comments

MaoZiming commented

Context

  • Pytorch version: 2.2.1+cu121
  • Operating System and version: Ubuntu 20.04.6 LTS

Your Environment

  • Installed using source? [yes/no]: no
  • Are you planning to deploy it using docker container? [yes/no]: no
  • Is it a CPU or GPU environment?: GPU (V100:1)
  • Which example are you using: examples/imagenet/main.py

Current Behavior

Segmentation fault (core dumped)
at output = model(images).

Steps to Reproduce

  1. python main.py -a resnet18 --dummy
  2. python main.py -a alexnet --dummy
MaoZiming commented

Pytorch installation/version issue.