linux或windows环境下pytorch的安设与检查验证

linux或windows环境下pytorch的安设与检查验证：

1、pytorch官网下载对应安装文件

https://pytorch.org/

在getstarted处选择对应版本

方法一：直接使用对应的安装指令进行安装

但是其实这样比较容易断或者出现runtimeout的error。解决办法：使用pip先进行源的查找，在开始下载后停掉，然后找到屏幕上打印出来的链接，手动下载whl 然后手动安装。

方法二：选择相应版本的whl进行离线下载

找到下载链接

进行whl文件的下载然后安装

pytorch从链接安装指定版本

pip3 install whlpip3 install torchvision

2、验证

import torchimport torchvisionprint(torch.cuda.is_available())#输出为true即可

3、pytorch版本查看

import torchprint(torch.__version__)

以上所述是小编给大家介绍的linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题)，希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。

（编辑：银川站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!