Python安装torch,transformers库

前言安装torch库安装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/

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

安装transformers库

同样的,使用pip安装transformers库

pip install transformers

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

pip install transformers==3.2.0

安装完后可以通过pip命令查看安装的版本

pip list

在这里插入图片描述

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

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

相关阅读

评论可见,请评论后查看内容,谢谢!!!
 您阅读本篇文章共花了: