pytorch环境创建及pytorch_geometric安装_pytorch geometric installation

一、pytorch环境创建

1.打开Anaconda Prompt,输入命令

conda create -n ‘name’ python=版本号

conda create -n pytorch python=3.8

进入环境

activate ‘name’

activate pytorch

2.打开网址Start Locally | PyTorch

选择对应版本的安装命令

例：pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

1.验证

输入

>>> import torch
 
>>> torch.cuda.is_available()

返回true则正确

2.查看版本

torch.__version__
 
torch.version.cuda

二、pytorch_geometric安装

进入官网，选择对应版本的链接

pytorch-geometric.com/whl/

下载对应版本的四个whl文件(cp38,代表python版本为3.8)

进入文件存放地址输入以下命令安装上述文件(注意文件名)

pip install torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl

pip install torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl

最后输入命令：

Pip install torch_geometric

pip install torch_geometric