安装Pytorch
1、查看本机的CUDA版本
如果输入nvidia-smi显示'nvidia-smi' 不是内部或外部命令,也不是可运行的程序
或批处理文件,则可以先输入【cd C:\Program Files\NVIDIA Corporation\NVSMI】再使用【nvidia-smi】命令。
2、PyTorch安装
进入官网:Get Started
这里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。
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 - CSDNhttps://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/
查看windows的CUDA版本
在3D设置中第三行即可查看CUDA最高可安装版本