参考链接: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
三.测试
查看conda包:
conda list
于是卸载,重新安装conda uninstall pytorch
conda install d:\download\pytorch-1.3.0-py3.7_cpu_0.tar.bz2
再测试
四.安装其他依赖包
作者要安装如下:
pip install jieba
pip install hanlp
pip install pytorch_pretrained_bert
五.下载中文预训练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安装好.
欢迎在评论区中进行批评指正,转载请注明来源,如涉及侵权,请联系作者删除。