问题描述:当我在jupyter-lab中import torchvision的时候,jupyter kernel died,网上查了半天资料发现torch和torchvision安装有问题,我想很多人也可能会碰到这种问题,因此,写下这篇内容。
原因:可能是需要安装cpu版本的包
解决办法:
第一步:中国大陆的朋友之下下面代码会加快安装速度。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
第二步:
conda install pytorch torchvision torchaudio cpuonly
参考:https://www.csdn.net/tags/MtTakg1sOTY0NTctYmxvZwO0O0OO0O0O.html
第三步:
重启一下jupyter-lab或jupyter哦,这步别忘了
至此解决了我的问题。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)