torch以及torch"/>
安装torch以及torch
1、torch分为CPU版和GPU版,官网安装链接Start Locally | PyTorch
(1)如图,安装CPU版,根据操作系统OS、语言环境Language、计算平台Compute Platform的真实情况选择,CPU版的Compute Platform选择CPU即可,复制官网给出的安装命令即可。
(2)安装GPU版,首先要查看电脑上的CUDA版本,在NVIDIA控制面板-系统信息-组件中可以看到,我的在更新前,CUDA版本是11.6,因此在选择Compute Platform时,按照上一张图片内容进行的选择,复制代码如下,也可以在选择过去的版本 Previous PyTorch Versions | PyTorch选择更为匹配的版本。
pip3 install torch torchvision torchaudio --extra-index-url
2、安装torch-geometric
按照anaconda下安装torch_geometric_F_aF_a的博客-CSDN博客_anaconda 安装 torch_geometric
安装即可。
pytorch的版本
确定pytorch的版本,最低1.4.0
python -c "import torch; print(torch.__version__)"
>>> 1.12.0
CUDA的版本
确定cuda的版本
python -c "import torch; print(torch.version.cuda)"
>>> 11.6
安装包
pip install torch-scatter -f ${TORCH}+${CUDA}.html
pip install torch-sparse -f ${TORCH}+${CUDA}.html
pip install torch-geometric
将其中${TORCH}和${CUDA}替换为第一二步确定的torch和cuda版本。
pip install torch-scatter -f .8.0+cu102.html
pip install torch-sparse -f .8.0+cu102.html
pip install torch-geometric
————————————————
以上部分内容未转载,原文链接:
3.如果安装过程中报错,可在网站 https://pytorch geometric选择与torch对应的版本,下载whl文件。
4.尤其要注意的一点时,安装torch-geometric时,一定要选择对应torch和cuda的版本,跟本机一致才能成功运行,否则会出现以下错误:
OSError: [WinError 127] 找不到指定的程序。 Error loading "D:\soft\anaconda\envs\pytorch\Lib\site-packages\torch\libcaffe2_detectron_ops.dll’ ,or one of its dependencies
更多推荐
安装torch以及torch
发布评论