安装torch以及torch

编程入门 行业动态 更新时间:2024-10-26 10:30:12

安装<a href=https://www.elefans.com/category/jswz/34/1755869.html style=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

本文发布于:2024-02-25 20:44:12,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1700243.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:torch

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!