pytorch和pyG安装

windows_python38_cuda11.1_pytorch_pyG安装

操作系统：windows10
显卡：GTX1650
CUDA版本11.1

  1. 下载安装CUDA Toolkit 11.1.0
  2. 新建conda环境，python3.8 conda create -n GNN python=3.8。激活conda activate GNN
  3. 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。
  4. 安装
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
image

posted @ 2023-03-01 19:40  shengtudai  阅读(190)  评论(0)  编辑  收藏  举报