Python安装torch，transformers库

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对应版本）