工欲善其事,必先利其器。开发工具的准备是进行深度学习的第一步。PyCharm是目前最流行的Python IDE。另外,搭建PyTorch环境,建议尽量利用GPU的算力,如果没有Nvidia显卡提供的GPU,CPU也可以,但示例代码的运算速度要慢很多。
PyTorch的前身便是Torch,Torch是纽约大学的一个机器学习开源框架,几年前在学术界非常流行,包括Lecun等大佬都在使用。但是由于使用的是一种绝大部分人没有听过的Lua语言,导致很多人都被吓退。后来随着Python的生态越来越完善,Facebook人工智能研究院推出了PyTorch并开源。PyTorch不是简单地封装Torch并提供Python接口,而是使用Python重新写了很多内容,不仅更加灵活,支持动态图,而且提供了Python接口。它是一个以Python优先的深度学习框架,不仅能够实现强大的GPU加速,同时还支持动态神经网络,这是很多主流深度学习框架(比如TensorFlow等)都不支持的。