场景:
今天在华为云上跑代码时执行代码:
from torchvision import models as models
model = models.resnet50(pretrained=True)
报错:ImportError:IProgress not found. Please update jupyter and ipywidgets.
原因:
是第一次导入resnet50预训练模型的时候报这个错。pretrained = True这个参数就是获得网络结构的同时,获得预训练的模型参数。需要下载。
这个问题本质上却少了相关的包,直接pip install IProgress或者是conda install IProgress并不能解决问题,其实这个报错后面有提示,去看官方文档就可以解决这个问题
解决方案:
官方文档说的很清楚:如果你的jupyter的内核和你运行代码的kernel不是一个环境的话,按底下这么做就行。
第一步,在base环境运行:
conda install -c conda-forge widgetsnbextension
第二步,进入你的虚拟环境并运行(py36是你虚拟环境的名字,自行替换):
source activate py36 #激活虚拟环境
conda install -c conda-forge ipywidgets
最后记得刷新一下,或更换内核再换回来。
再在华为云上运行上述代码时就不会报错了。