PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序。
PyTorch的前身是Torch,其底层和Torch框架一样,但是使用Python重新写了很多内容,不仅更加灵活,支持动态图,而且提供了Python接口。它是由Torch7团队开发,是一个以Python优先的深度学习框架,不仅能够实现强大的GPU加速,同时还支持动态神经网络。
PyTorch既可以看作加入了GPU支持的numpy,同时也可以看成一个拥有自动求导功能的强大的深度神经网络。除了Facebook外,它已经被Twitter、CMU和Salesforce等机构采用。
anaconda这是一个非常常用的python包集成管理工具,其中预安装了很多python库,使得我们不需要去手动安装各种的第三方库,我们知道自己取手动安装的过程中,很容易就会遇到一些报错,解决起来也非常的麻烦。
anaconda官网:https://www.anaconda.com/products/distribution#Downloads
下载完软件包一路点击安装就行了,安装成功后会出现如下界面。
查看是否安装成功命令:conda --version
pytorch官方文档:https://pytorch.org/
点击install进入安装界面选择自己对应python版本的安装命令:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
果断输入y,之后就是漫长的等待。
1、输入python,进入python环境
2、输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了
3、接下来,输入 torch.cuda.is_available,如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU
可怜的我没GPU这东西。