linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题)
1、pytorch官网下载对应安装文件
在getstarted处选择对应版本
方法一:直接使用对应的安装指令进行安装
但是其实这样比较容易断或者出现runtimeout的error。解决办法:使用pip先进行源的查找,在开始下载后停掉,然后找到屏幕上打印出来的链接,手动下载whl 然后手动安装。
方法二:选择相应版本的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常用命令的用法(二)————文本编辑器命令vi/vim
这篇文章主要介绍了Linux常用命令的用法(二)————文本编辑器命令vi/vim,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2019-04-04在Ubuntu 14 上安装 Nginx-RTMP 流媒体服务器的教程
RTMP流媒体协议是 一套 Adobe 开发的音频视频实时传输协议;nginx-rtmp 是一个基于nginx的 RTMP服务模块,开源,免费,接下来通过本文给大家介绍在Ubuntu 14 上安装 Nginx-RTMP 流媒体服务器,需要的朋友可以参考下2019-11-11
最新评论