当前位置:   article > 正文

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

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.验证

输入

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

返回true则正确

2.查看版本

  1. torch.__version__
  2. 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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/860466
推荐阅读
相关标签
  

闽ICP备14008679号