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
Python安装torch,transformers库_第1张图片
Python安装torch,transformers库_第2张图片

Python安装torch,transformers库_第3张图片

安装transformers库

同样的,使用pip安装transformers库

pip install transformers

Python安装torch,transformers库_第4张图片
不指定版本的话,默认安装的是最高版本,但是高版本的兼容性一般不太好,这里我们也可以自己指定一个transformers安装。

pip install transformers==3.2.0

Python安装torch,transformers库_第5张图片
安装完后可以通过pip命令查看安装的版本

pip list

在这里插入图片描述
Python安装torch,transformers库_第6张图片

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

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

你可能感兴趣的:(Python学习笔记,python,深度学习,pytorch,transformer)