当前位置:   article > 正文

安装Pytorch、pytorch_transformers_pycharm pytorch transformers 入门安装

pycharm 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最高可安装版本

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

闽ICP备14008679号