赞
踩
一、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安装
进入官网,选择对应版本的链接
下载对应版本的四个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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。