当前位置:   article > 正文

pytorch安装_install pytorch

install pytorch

一、pytorch介绍

PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理应用程序。PyTorch既可以看作加入了GPU支持的numpy,同时也可以看成一个拥有自动求导功能的强大的深度神经网络。除了Facebook外,它已经被Twitter、CMU和Salesforce等机构采用

二、安装pytorch

第一步:成功安装Anconda

第二步:安装pytorch

网上好多都是GPU、CUDA(CUDNN)安装教程,而且还要求是英伟达的显卡(NV),而查询我的电脑显卡为Intel(R) Iris(R) Xe Graphics,所以安装步骤稍微不同。

首先进入pytorch官网:Start Locally | PyTorch,选择相对应的版本,自动生成安装命令,并在cmd中输入安装命令:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

2.1在cmd中输入进行安装

2.2测试是否安装成功

输入下列代码

  1. import torch
  2. import torchvision
  3. print(torch)

出现<module 'torch' from 'D:\\MySoftware\\anaconda3\\lib\\site-packages\\torch\\__init__.py'>

即为安装成功!!!! 

2.3测试能否正常使用

打开jupyter notebook 写代码,看是否能导入torch包,如果不能,可以看下一步

3.1修复方法

在jupyter notebook环境中安装一个pytorch环境

1、打开anaconda,输入conda activate Pytorch进入pytorch环境,如果你的pytorch文件名不是pytorch,改为你的文件名

2、输入conda install ipython,在pytorch环境中安装ipython,如果下载的比较慢,可以换源

conda install ipython

换源方法(最常用的方法:利用anaconda)

打开 Anaconda Prompt

在命令行输入以下命令,即可

(注意:每次只能输入一行代码)

  1. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  2. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  3. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  4. conda config --set show_channel_urls yes

3、下载完之后,输入conda install jupyter,在pytorch环境中下载jupyter

conda install jupyter

4、下载完之后,输入python -m ipykernel –user –name pytorch(pytorch可以改为任意名字,但是不能含有中文)

python -m ipykernel –user –name pytorch

5、下载完成之后,打开jupyter notebook

这边会多出来一个pytorch名称,也是我刚刚命名的名称

6、点击可以进入pytorch环境,编辑自己的代码

这就是我的安装步骤哦,希望能给努力学习的你提供一点帮助!!!

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

闽ICP备14008679号