本文介绍: pytorch拥有CPU和GPU两个版本,其中毋庸置疑的是如果使用GPU的话,速度往往超于CPU版本。这里呢,我将就【pytorch-gpu版本安装】这一问题,进行解决,主要以检查GPU的cuda版本以及检查当前pytorch的版本、安装pytorch-gpu版本来具体描述
前言
PyTorch是一款广泛使用的python深度学习框架,它能够帮助研究者们快速构建和训练复杂的神经网络,在人工智能领域无疑是【宠儿】的存在。
但刚进门的小白们应该都有困惑:为什么有些基于pytorch框架吃CPU,油得却是吃GPU,这里呢,主要是因为pytorch拥有CPU和GPU两个版本,其中毋庸置疑的是如果使用GPU的话,速度往往超于CPU版本。
这里呢,我将就【pytorch-gpu版本安装】这一问题,进行解决,主要以检查GPU的cuda版本以及检查当前pytorch的版本、安装pytorch-gpu版本来具体描述。
请注意:这里我给出的方案全部基于windows系统、nvidia显卡,其他系统仅供参考
环境准备及安装
首先呢,你需要了解两个非常重要的问题:
1、你有没有GPU
2、安装pytorch-gpu版本
3、检查pytorch版本
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。