jupyter notebook默认使用base环境,但我的torch装在虚拟环境中,于是需要把虚拟环境添加上。
参考文章:https://www.jb51.net/article/184447.htm
添加虚拟环境需要用的ipykernel这个包,如果无,第一步是装上ipykernel
pip install ipykernel
假设要添加的虚拟环境是XX,展示的名称是YY
python -m ipykernel install --name XX --display-name "YY"
重启jupyter notebook就可以看见添加的虚拟环境
选中后右上角状态显示正常了就可以使用了