pytorch和pyG安装
windows_python38_cuda11.1_pytorch_pyG安装
操作系统:windows10
显卡:GTX1650
CUDA版本11.1
- 下载安装CUDA Toolkit 11.1.0
- 新建conda环境,python3.8
conda create -n GNN python=3.8
。激活conda activate GNN
- win下CUDA11.1能支持GPU的版本:
pytorch==1.8.1+cu111
,torchvision==0.9.1+cu111
,torchaudio==0.8.1
,torch_scatter==2.0.8
,torch_sparse==0.6.12
,torch_cluster==1.5.9
,torch_spline_conv==1.2.1
。 - 安装
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
pip install torch_scatter==2.0.8 torch_sparse==0.6.12 torch_cluster==1.5.9 torch_spline_conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html
pip install torch-geometric
其他版本要再尝试了,截至2023-03-01,这是windows上能安装的最新版本。
更新
再pycharm中使用conda环境下的pyG,会一直updating skeleton。需要换成python的Virtualenv