赞
踩
这个是Pytorch安装的步骤
conda是一个包管理器,安装的时候不需要安装cuda,在安装Pytorch的时候Cuda就已经安装好了
验证安装:
GPU是一种处理特殊计算的处理器,它比CPU速度更快,并行计算就可以借助GPU,也就是说最适合GPU解决的问题是可以并行完成的任务。而神经网络就是并行,所以GPU非常适合神经网络的计算。
CUDA正好和GPU对接,更快的加速我们训练的过程,所以说使用CUDA就必须有GPU
下面这个例子,首先我们先创建了一个tensor张量,随后我们把张量tensor放到CUDA中进行加速,我们可以看到输出除了tensor之外还有标明设备的标识,device就是指明那个设备,因为Pytorch支持多个GPU,所以这个的0是个索引,代表0号GPU
GPU对于那些可以被分解为许多小任务的任务来说效果很好,如果计算任务已经足够小了,把它分解并且移动到GPU上面,我们不会有太大的收获。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。