Jupyter notebook中怎么添加Pytorch运行环境
作者: Jintao
日期: 2021-7-9
简介: 本文内容为了解决用户在使用Anaconda3 里面的Jupyter Notebook 遇到 import torch报错的情况 [图1] 应该如何解决的方法, 在此分享给大家,希望对大家有用!
[图1] - import torch 报错的情况
- 出现像类似图1 的错误时,说并Python 并没有下载PyTorch拓展包,需要从官网进行下载
- 下载链接: https://pytorch.org/ [在 2.0 内容介绍]
- 以下内容专门针对第一次下载Anaconda 的用户,然后需要下载PyTorch来运行torch的代码块
1.0 打开Anaconda.Navigator 界面
- 打开Anaconda.Navigator 界面 (如下图), 然后点击运行 CMD.exe Prompt 0.1.1 (我画黑色圈的地方)。 之后会弹出一个终端对话框。
2.0 下载PyTorch
- 完成1.0弹出对话框之后, 在此对话框输入下载指令
- 下载链接: https://pytorch.org/
- 往下拉会看见下载指令会根据你的系统以及下载平台有关,我是Win10系统,然后我的笔记本自带一个NAVDIA显卡,所以我我默认选择 Stable + Windows + Conda + Python + CUDA 10.2
- 你们也可以根据你们的电脑的配置和需求来进行调整
- 配置完之后,在最下方会自动显示一串代码,复制这串代码到之前在1.0运行的终端对话框,并且运行。
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
3.0 等待对话框回应,等待出现以下类似界面时
- 若在下载时出现报错现象,则很有可能是因为你之前曾经下过一些内容,然后是包含在我们现在准备要下载的文件里,因此出现了重复下载的现象。遇到这种情况最好的办法一般是先卸载你之前所下载的内容。比如以下卸载numpy先。
- pip uninstall numpy
- 输入 y 确认下载PyTorch
- 若下载到一半之后中断了的话不用担心的,可以再次运行刚才的下载代码,然后系统会自动识别已经下载的内容,然后继续下载还未下载的内容。小编也是恰好遇到断网,然后报错说我网络未识别,需要重新下载
3.0 等待对话框回应
- 当出现自动提示信息时 C:\Users\Danie\> 表示PyTorch 已经下载完成。用户可以在此页面进行测试,如果 import torch 没有报错的话那就是已经下载好了