当前位置:   article > 正文

Python安装torch,transformers库_pip install transformers

pip install transformers

前言

最近由于竞赛需要安装Python中机器学习torch已经transformers库函数,但是找了很多帖子,并没有找到一种满意的解决方案。在这里分享一下最终的解决方法。
安装两个库函数的网址

Python环境:3.8
windows系统:win10系统

注意:安装这两个库之前,需要首先安装numpy库

安装torch库

通过cmd命令安装

pip install torch==1.4.0 torchvision==0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/
  • 1

其中torch版本和torchvision版本根据自己电脑的实际情况选择,后面的链接为清华镜像链接,也可以改为国内其他镜像链接如阿里巴巴镜像
我的Python版本为3.8。这里选择torch版本为1.4.0,torchvision版本为0.5.0
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

安装transformers库

同样的,使用pip安装transformers库

pip install transformers
  • 1

在这里插入图片描述
不指定版本的话,默认安装的是最高版本,但是高版本的兼容性一般不太好,这里我们也可以自己指定一个transformers安装。

pip install transformers==3.2.0
  • 1

在这里插入图片描述
安装完后可以通过pip命令查看安装的版本

pip list
  • 1

在这里插入图片描述
在这里插入图片描述

大家如果对于安装还有其他问题,可以看以下两个推荐链接,写的很好,希望可以帮到你

保姆级别的Anaconda/pip高速安装pytorch1.7.1
Python安装torch(含torch和torchvision对应版本)

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

闽ICP备14008679号