配置github项目ccks_kg

参考链接:Windows下安装pytorch1.3.0安装以及中途遇到的安装问题

一.首先要安装虚拟环境

conda create -n ccks_kg python=3.7

conda activate ccks_kg

二.在清华镜像下载win64对应的pytorch1.3版本安装文件

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/ 地址链接

因为我不知道是否是GPU版本,并且我的GPU是CUDA11,所以直接安装的CPU版本,也就是 pytorch-1.3.0-py3.7_cpu_0.tar.bz2 ,下载到本地D:\download。

然后安装

conda install d:\download\pytorch-1.3.0-py3.7_cpu_0.tar.bz2

然后手动安装依赖包(如果我们直接用conda安装的话,会自动下载安装,但是我们是离线安装的):

conda install numpy pyyaml mkl cmake cffi

三.测试

image-20211001232928880

查看conda包:

conda list

image-20211001232941237

于是卸载,重新安装
conda uninstall pytorch
conda install d:\download\pytorch-1.3.0-py3.7_cpu_0.tar.bz2
再测试

image-20211001233015455

四.安装其他依赖包

作者要安装如下:

pip install jieba
pip install hanlp
pip install pytorch_pretrained_bert

image-20211001233043994

五.下载中文预训练BERT模型

[下载中文预训练BERT模型 (pytorch) bert-base-chinese](https://blog.csdn.net/sdaujz/article/details/107547503 “下载中文预训练BERT模型 (pytorch) bert-base-chinese”) 将config.json,pytorch_model.bin,vocab.txt三个文件下载后*./bert-base-chinese*文件下.

六.重新安装pytorch 执行main.py报错,说是pytorch用不了GPU,执行下列代码版本号为1.7.1+CPU,false.于是继续下载pytorch和cuda相匹配. “python import torch print(torch.__version__) print(torch.cuda.is_available()) “ 然后我又去官网, 复制了cuda11的pip命令进行安装,安装失败, 复制conda安装,安装成功,再执行上段代码,之后再执行main.py正常执行.

七.总结 1. 我觉得首先要看官方的readme,确保自己所需要的包是否都下载成功. 2. 一定要把pytorch和cuda对应的版本弄好,还有cudann安装好.


欢迎在评论区中进行批评指正,转载请注明来源,如涉及侵权,请联系作者删除。

×

喜欢就点赞,疼爱就打赏