自己记录以下
完整复现模型模型ner
conda info -e
conda activate name
conda install python==3.8.16
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
python
>>> import torch
>>> torch.__version__
>>>torch.cuda.is_available()
确定torch和cuda
已安装:查看torch和cuda版本
import torch
print(torch.__version__) #注意是双下划线
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())
未安装:安装我的版本号
# CUDA 11.3
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
安装好检查一下版本号
我的如下:
操作系统:Win10
Cuda版本:Cuda 11.3
Pytorch版本:Pytorch 1.12.0
Python版本:Python 3.8.13
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cu113.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.12.0+cu113.html
pip install torch-cluster -f https://data.pyg.org/whl/torch-1.12.0+cu113.html
pip install torch-spline-conv -f https://data.pyg.org/whl/torch-1.12.0+cu113.html
pip install torch-geometric
如果这一步无法下载,超时等问题
直接点击这个链接,进行下载:https://pytorch-geometric.com/whl/torch-1.12.0+cu113.html
验证以下:
python -c "import torch_geometric"
出错
解决方法
pip install torch-geometric -i http://pypi.douban.com/simple --trusted-host pypi.douban.com