jupyter中torch库的安装与虚拟环境的搭建

创建环境

  1. 打开Anaconda Prompt(在开始菜单中找到Anaconda文件夹,并打开Anaconda Prompt)


  2. 创建一个新的虚拟环境,例如名为“pytorch”的环境,并指定Python版本为3.7或3.9。你可以使用以下命令创建环境:conda creat --name pytorch_env python=3.9


    代码:conda create -n pytorch python=3.9
  3. 配置清华的PyTorch镜像源


    代码:  

    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/conda config --set show_channel_urls yes

  4. 激活新创建的环境


    代码:conda activate pytorch

查找自己电脑对应的CUDA版本相匹配的PyTorch版本。(重要)

  1. 看你的电脑上CUDA的版本。在Anaconda Prompt中输入以下命令:nvidia-smi(右上角CUDA就是我们查找的数据11.6


    代码:nvidia-smi

  2. 在pytroch官网(PyTorch)上查找对应的PyTorch版本


  3. 点击左下角的Previous versions of PyTorch(找到对应的)


下载等待及问题

  1. 将上面代码输入后的反馈:


  2. 后面输入y后进入下载等待:


  3. 后面漫长,如果中途退出,不会报错,可以重新进入环境,继续下载,输入2中的代码:


检验并安装内核

  1. 检验pytorch环境是否安装完毕:


    代码:conda activate pytorch

  2. 安装内核:安装ipykernel,它是一个可以让你管理 IPython 内核的软件包


    代码:pip install ipykernel

  3. 中途可能会出现一些红色报错,问题出在你试图安装ipykernel时,pip的依赖解析器发现matplotlib有几个依赖包没有被安装,它们分别是cycler、fonttools和kiwisolver。matplotlib是一个绘图库,它依赖这些包来正常工作。但内核已经安装完成。用pip install安装即可。


  4. 如果你之前安装过ipykernel想要改变它的位置,可以进行如下操作,再次安装。


    代码:jupyter kernelspec uninstall mykernel

  5. 如果你要检查是否安装成功,可以使用pip list


将内核与python绑定

  1. 然后将内核与python进行绑定。通常我们不会使用 ipkernel 这个名称来创建自定义内核,因为 ipykernel 是 ipython 项目的官方内核。可以选择一个更有描述性的名称来代替 mykernel,比如 myenvkernel,其中 myenv 是你的虚拟环境的名称。


    代码:python -m ipykernel install --user --name=myenvkernel

  2. 打开你的jupyter Notebook创建一个新的python文件


  3. 点击右上角的Python3(我的这里是Python3(ipykernel)),将环境更换为myenvkernel


  4. 最后运行import torch看是否能成功调用该库

  • 48
    点赞
  • 55
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值