search

aws / sagemaker-experiments

Experiment tracking and metric logging for Amazon SageMaker notebooks and model training.
Apache License 2.0
125 stars 36 forks source link

mnist-handwritten-digits-classification-experiment.ipynb - No module named 'torchvision' #117

Closed adambl closed 2 years ago

adambl commented 3 years ago

Describe the bug A clear and concise description of what the bug is. Getting error - ModuleNotFoundError: No module named 'torchvision' when running mnist-handwritten-digits-classification-experiment.ipynb To Reproduce Steps to reproduce the behavior: upload the notebook as specified in the "Track and Compare Tutorial" Expected behavior A clear and concise description of what you expected to happen. The notebook should run without errors Screenshots If applicable, add screenshots to help explain your problem. Screen Shot 2021-02-04 at 17 57 57

Environment: Framework (e.g. TensorFlow) / Algorithm (e.g. KMeans):PyTorch Framework Version: Python Version:Python 3.7.6 CPU or GPU:CPU Python SDK Version: Are you using a custom image: No

Additional context Add any other context about the problem here.

jerrypeng7773 commented 3 years ago

Hi, please run the cell below Install PyTroch section, then the torchvision will be installed.

https://github.com/aws/amazon-sagemaker-examples/blob/master/sagemaker-experiments/mnist-handwritten-digits-classification-experiment/mnist-handwritten-digits-classification-experiment.ipynb