网站建设资讯

NEWS

网站建设资讯

linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题)

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

创新互联专业提供成都主机托管四川主机托管成都服务器托管四川服务器托管,支持按月付款!我们的承诺:贵族品质、平民价格,机房位于中国电信/网通/移动机房,雅安服务器托管服务有保障!

https://pytorch.org/

在getstarted处选择对应版本

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

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

linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题)

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

找到下载链接

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

pytorch从链接安装指定版本

pip3 install whl
pip3 install torchvision

2、验证

import torch
import torchvision
print(torch.cuda.is_available())#输出为true即可

3、pytorch版本查看

import torch
print(torch.__version__)

总结

以上所述是小编给大家介绍的linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题),希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对创新互联网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!


名称栏目:linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题)
文章出自:http://cdweb.net/article/jjcedj.html