安装环境:conda 虚拟环境
!!!踩坑安装流程!!!(不要照以下内容操作!!!)
在torch官网进行conda安装
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch
在pyg库进行安装
conda install pyg -c pyg
报错:OSError: …/torch_sparse/_version_cuda.so: undefined symbol: _ZN5torch3jit17parseSchemaOrNameERKSs
原因:因为conda安装pytorch的时候会同时安装CPU和GPU版本,即便是安装时指定了具体版本的GPU版,所以建议使用whl进行安装,
换了一种安装方法:
前置条件:已经安装完nvidia驱动,这部分参考别的blog安装即可
首先:
conda create -n your_name python=3.7
conda activate your_name
conda config --append channels conda-forge
torch 官网安装,用whl
pyg 官网照whl安装
安装成果:成功
总结:whl 很好用!不要相信conda自动安装,一定要自己指定安装!!!