当前位置:   article > 正文

Pytorch详细教程——3.PyTorch Install and CUDA_安装pytorch的时候会安装cuda吗

安装pytorch的时候会安装cuda吗

3. PyTorch Install - Quick And Easy

这个是Pytorch安装的步骤
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
conda是一个包管理器,安装的时候不需要安装cuda,在安装Pytorch的时候Cuda就已经安装好了

验证安装:

在这里插入图片描述
在这里插入图片描述

4. CUDA Explained - Why Deep Learning Uses GPUs

GPU是一种处理特殊计算的处理器,它比CPU速度更快,并行计算就可以借助GPU,也就是说最适合GPU解决的问题是可以并行完成的任务。而神经网络就是并行,所以GPU非常适合神经网络的计算。

CUDA正好和GPU对接,更快的加速我们训练的过程,所以说使用CUDA就必须有GPU

下面这个例子,首先我们先创建了一个tensor张量,随后我们把张量tensor放到CUDA中进行加速,我们可以看到输出除了tensor之外还有标明设备的标识,device就是指明那个设备,因为Pytorch支持多个GPU,所以这个的0是个索引,代表0号GPU
在这里插入图片描述

GPU对于那些可以被分解为许多小任务的任务来说效果很好,如果计算任务已经足够小了,把它分解并且移动到GPU上面,我们不会有太大的收获。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/正经夜光杯/article/detail/813546
推荐阅读
相关标签
  

闽ICP备14008679号