安装Pytorch、pytorch_transformers

安装Pytorch

1、查看本机的CUDA版本

如果输入nvidia-smi显示'nvidia-smi' 不是内部或外部命令,也不是可运行的程序
或批处理文件,则可以先输入【cd C:\Program Files\NVIDIA Corporation\NVSMI】再使用【nvidia-smi】命令。

2、PyTorch安装

进入官网:Get Started

PyTorchicon-default.png?t=N7T8https://pytorch.org/

这里Pytorch版本的选择,我选择的是Stable,Windows,Conda,Python,CUDA 10.2。最后的的Compute Platform根据大家的需求来定。大家如果想在自己电脑(具有NVIDIA显卡)上跑通代码,就选CUDA,如果不需要在自己电脑上跑(在服务器上跑)或者没有独立显卡,就选CPU。

由于我的NAVIDA的CUDA版本为11.0,这里选择比自己版本低的CUDA。我只能选择10.2,因为11.3有点高了。但pytorch官网只支持11.3,10.2的不支持了,无法下载,所以我找了之前的版本,使用了以下代码进行安装。

【Previous PyTorch Version】

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

 输入y确认,然后回车。

此过程很慢很慢,需要耐心等待,安装完成后会提示:done。

输入pip list,可以看到安装成功。

 输入python进入下Python环境,然后输入import torch,如果没有报错说明可以导入成功。

输入torch.cuda.is_available()查看torch是否可以使用显卡,True就代表可以,False代表不可以 T_T。

参考:(48条消息) Python安装Pytorch教程(图文详解)_振华OPPO的博客-CSDN博客_python安装pytorchicon-default.png?t=N7T8https://blog.csdn.net/qq_42257666/article/details/121361983

3.pytorch_transformers安装

直接使用以下代码。

pip install pytorch_transformers -i https://pypi.douban.com/simple

anconda导入pycharm

Settings

Python Interpreter中Add 

Conda Environment中选择Existing environment

选择 Interpreter

OK即可!

 参考:(49条消息) 怎么把anaconda导入pycharm - CSDNicon-default.png?t=N7T8https://www.csdn.net/tags/MtTaEgzsNzM4ODAwLWJsb2cO0O0O.html

或者可以右下角直接选择添加【Add New Interpreter】

 安装spacy

 运行代码出现错误can't find model 'en_core_web_sm'

https://github.com/explosion/spacy-models/releases/tag/en_core_web_trf-3.0.0a1

直接从github上下载en_core_web_sm ,手动进行安装。

pip install 语言模型的绝对路径 -i https://pypi.tuna.tsinghua.edu.cn/simple/

参考:
https://blog.csdn.net/qq_45520647/article/details/123536812icon-default.png?t=N7T8https://blog.csdn.net/qq_45520647/article/details/123536812

查看windows的CUDA版本

 

在3D设置中第三行即可查看CUDA最高可安装版本

  • 21
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值