当前位置:   article > 正文

jupyter切换pytorch_gpu内核_jupyter可以切换gpu和cpu吗

jupyter可以切换gpu和cpu吗

疑问:

  • 之前一直使用的tensorflow,使用jupyter切换python内核方法生成的内核时可以使用的
  • 但是今天用同样的方法生成pytorch内核的时候,在jupyterlab里面却不能顺利的使用:
import torch

以下是解决方法:

  • 删除用之前方式添加的内核
  1. (CMD)查看内核安装的位置 jupyter kernelspec list
  2. (CMD)jupyter kernelspec remove 你的内核名称
  • 安装nb_conda_kernels包,自动把虚拟环境下的python生成内核,直接就可以在jupyterlab中看到
conda install nb_conda_kernels

效果:

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/711289
推荐阅读
相关标签
  

闽ICP备14008679号